explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-22 22:40:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix import of 'cyclic_triangular' from thinc

Browse Source
This commit is contained in:
Matthw Honnibal 2019-11-19 11:17:07 +01:00
parent e675609e27
commit 0530db9504
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/cli/pretrain.py
View File

@ -14,7 +14,7 @@ from thinc.neural.util import require_gpu
from wasabi import Printer
import srsly
from thinc.neural.util import to_categorical
from thinc.rates import cyclic_triangular_rate
from thinc.rates import cyclic_triangular
from ..errors import Errors
from ..tokens import Doc
@ -134,6 +134,7 @@ def pretrain(
if isinstance(config[key], Path):
config[key] = str(config[key])
util.fix_random_seed(seed)
msg = Printer()
if gpu_id != -1:
has_gpu = require_gpu(gpu_id=gpu_id)
msg.info("Using GPU {}".format(gpu_id))
@ -234,7 +235,7 @@ def pretrain(
min_lr = optimizer.alpha / 3
max_lr = optimizer.alpha * 2
period = 10000
learn_rates = cyclic_triangular_rate(min_lr, max_lr, period)
learn_rates = cyclic_triangular(min_lr, max_lr, period)
for epoch in range(epoch_start, n_iter + epoch_start):
for batch_id, batch in enumerate(
util.minibatch_by_words(((text, None) for text in texts), size=batch_size)