mirror of https://github.com/explosion/spaCy.git
Don't call begin_training if updating new model (see #3059) [ci skip]
This commit is contained in:
parent
ab4b61fb6e
commit
e3405f8af3
|
@ -56,6 +56,9 @@ def main(model=None, output_dir=None, n_iter=100):
|
|||
# get names of other pipes to disable them during training
|
||||
other_pipes = [pipe for pipe in nlp.pipe_names if pipe != "ner"]
|
||||
with nlp.disable_pipes(*other_pipes): # only train NER
|
||||
# reset and initialize the weights randomly – but only if we're
|
||||
# training a new model
|
||||
if model is None:
|
||||
optimizer = nlp.begin_training()
|
||||
for itn in range(n_iter):
|
||||
random.shuffle(TRAIN_DATA)
|
||||
|
@ -68,7 +71,6 @@ def main(model=None, output_dir=None, n_iter=100):
|
|||
texts, # batch of texts
|
||||
annotations, # batch of annotations
|
||||
drop=0.5, # dropout - make it harder to memorise data
|
||||
sgd=optimizer, # callable to update weights
|
||||
losses=losses,
|
||||
)
|
||||
print("Losses", losses)
|
||||
|
|
Loading…
Reference in New Issue