From 2b26c007cda336c797cd235732676ed471d53b99 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 29 Apr 2018 14:09:02 +0000 Subject: [PATCH] Revert "Disable batch size compounding in ud-train" This reverts commit 8a120fb4553684123fe280d3e6cd8a99cfe81c21. --- spacy/cli/ud_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/cli/ud_train.py b/spacy/cli/ud_train.py index 58d46e34a..853cff9b3 100644 --- a/spacy/cli/ud_train.py +++ b/spacy/cli/ud_train.py @@ -338,7 +338,7 @@ def main(ud_dir, parses_dir, config, corpus, limit=0, use_gpu=-1): optimizer = initialize_pipeline(nlp, docs, golds, config, use_gpu) - batch_sizes = compounding(config.batch_size, config.batch_size, 1.001) + batch_sizes = compounding(config.batch_size//10, config.batch_size, 1.001) for i in range(config.nr_epoch): docs = [nlp.make_doc(doc.text) for doc in docs] Xs = list(zip(docs, golds))