explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-05-31 03:03:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Test and fix #1919: Error resuming training

Browse Source
This commit is contained in:
Matthew Honnibal 2018-02-02 02:32:40 +01:00
parent 6b1126c312
commit f74a802d09
2 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
spacy/syntax/nn_parser.pyx
View File

@ -827,6 +827,7 @@ cdef class Parser:
for action, labels in actions.items(): for action, labels in actions.items():
for label in labels: for label in labels:
self.moves.add_action(action, label) self.moves.add_action(action, label)
cfg.setdefault('token_vector_width', 128)
if self.model is True: if self.model is True:
cfg['pretrained_dims'] = self.vocab.vectors_length cfg['pretrained_dims'] = self.vocab.vectors_length
self.model, cfg = self.Model(self.moves.n_moves, **cfg) self.model, cfg = self.Model(self.moves.n_moves, **cfg)
@ -836,11 +837,12 @@ cdef class Parser:
self.model[1].ops.allocate((5, cfg['token_vector_width']))) self.model[1].ops.allocate((5, cfg['token_vector_width'])))
self.init_multitask_objectives(gold_tuples, pipeline, sgd=sgd, **cfg) self.init_multitask_objectives(gold_tuples, pipeline, sgd=sgd, **cfg)
link_vectors_to_models(self.vocab) link_vectors_to_models(self.vocab)
self.cfg.update(cfg) else:
elif sgd is None: if sgd is None:
sgd = self.create_optimizer() sgd = self.create_optimizer()
self.model[1].begin_training( self.model[1].begin_training(
self.model[1].ops.allocate((5, cfg['token_vector_width']))) self.model[1].ops.allocate((5, cfg['token_vector_width'])))
self.cfg.update(cfg)
return sgd return sgd
def add_multitask_objective(self, target): def add_multitask_objective(self, target):

10
spacy/tests/regression/test_issue1919.py Normal file
View File

@ -0,0 +1,10 @@
'''Test that nlp.begin_training() doesn't require missing cfg properties.'''
from __future__ import unicode_literals
import pytest
from ... import load as load_spacy
@pytest.mark.models('en')
def test_issue1919():
nlp = load_spacy('en')
opt = nlp.begin_training()