Fix name of update_shared flag

This commit is contained in:
Matthew Honnibal 2017-08-20 18:19:06 +02:00
parent 8a59718fd6
commit f2f9229964
1 changed files with 1 additions and 1 deletions

View File

@ -92,7 +92,7 @@ def train(cmd, lang, output_dir, train_data, dev_data, n_iter=20, n_sents=0,
docs, golds = zip(*batch) docs, golds = zip(*batch)
nlp.update(docs, golds, sgd=optimizer, nlp.update(docs, golds, sgd=optimizer,
drop=next(dropout_rates), losses=losses, drop=next(dropout_rates), losses=losses,
update_tensors=True) update_shared=True)
pbar.update(sum(len(doc) for doc in docs)) pbar.update(sum(len(doc) for doc in docs))
with nlp.use_params(optimizer.averages): with nlp.use_params(optimizer.averages):