This commit is contained in:
Richard Liaw 2020-06-22 19:01:22 -07:00
parent 0b9113a1da
commit b75b636d59

View File

@ -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):