From b75b636d5909fbc7aff45ca28a2119f4bc6f786d Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Mon, 22 Jun 2020 19:01:22 -0700 Subject: [PATCH] revert --- spacy/cli/train_from_config.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/spacy/cli/train_from_config.py b/spacy/cli/train_from_config.py index 1a66478bc..dc8e8fba2 100644 --- a/spacy/cli/train_from_config.py +++ b/spacy/cli/train_from_config.py @@ -1,5 +1,6 @@ from typing import Optional, Dict, List, Union, Sequence from timeit import default_timer as timer + import srsly from pydantic import BaseModel, FilePath import plac @@ -520,6 +521,7 @@ def train_while_improving( raw_batches = util.minibatch( (nlp.make_doc(rt["text"]) for rt in raw_text), size=8 ) + for step, batch in enumerate(train_data): dropout = next(dropouts) with nlp.select_pipes(enable=to_enable):