mirror of
https://github.com/explosion/spaCy.git
synced 2025-07-11 00:32:40 +03:00
Initialize fine tuning to 0.5
This commit is contained in:
parent
62878e50db
commit
c10f63bf10
|
@ -388,7 +388,7 @@ def fine_tune(embedding, combine=None):
|
||||||
return output, fine_tune_bwd
|
return output, fine_tune_bwd
|
||||||
model = wrap(fine_tune_fwd, embedding)
|
model = wrap(fine_tune_fwd, embedding)
|
||||||
model.mix = model._mem.add((model.id, 'mix'), (1,))
|
model.mix = model._mem.add((model.id, 'mix'), (1,))
|
||||||
model.mix.fill(0.0)
|
model.mix.fill(0.5)
|
||||||
model.d_mix = model._mem.add_gradient((model.id, 'd_mix'), (model.id, 'mix'))
|
model.d_mix = model._mem.add_gradient((model.id, 'd_mix'), (model.id, 'mix'))
|
||||||
return model
|
return model
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user