mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-26 01:46:28 +03:00
Remove b1 and b2 decay
This commit is contained in:
parent
83ac227bd3
commit
fb56028476
|
@ -192,11 +192,7 @@ def train(
|
|||
# Start with a blank model, call begin_training
|
||||
optimizer = nlp.begin_training(lambda: corpus.train_tuples, device=use_gpu)
|
||||
|
||||
optimizer.b1_decay = 0.0001
|
||||
optimizer.b2_decay = 0.0001
|
||||
nlp._optimizer = None
|
||||
optimizer.b1_decay = 0.003
|
||||
optimizer.b2_decay = 0.003
|
||||
|
||||
# Load in pre-trained weights
|
||||
if init_tok2vec is not None:
|
||||
|
|
Loading…
Reference in New Issue
Block a user