From f5b1ad41002eecd66feb244d51c60b57c4a88db2 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 28 Jan 2018 21:00:32 +0100 Subject: [PATCH] Limit parser model size, to hopefully reduce memory during CI tests --- spacy/tests/parser/test_beam_parse.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/tests/parser/test_beam_parse.py b/spacy/tests/parser/test_beam_parse.py index 59e307bcb..855b2e1e9 100644 --- a/spacy/tests/parser/test_beam_parse.py +++ b/spacy/tests/parser/test_beam_parse.py @@ -17,7 +17,7 @@ def test_beam_parse(): nlp = Language() nlp.add_pipe(DependencyParser(nlp.vocab), name='parser') nlp.parser.add_label('nsubj') - nlp.begin_training() + nlp.parser.begin_training([], token_vector_width=8, hidden_width=8) doc = nlp.make_doc(u'Australia is a country') nlp.parser(doc, beam_width=2)