mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-26 01:46:28 +03:00
Limit parser model size, to hopefully reduce memory during CI tests
This commit is contained in:
parent
6d978e5c35
commit
f5b1ad4100
|
@ -17,7 +17,7 @@ def test_beam_parse():
|
||||||
nlp = Language()
|
nlp = Language()
|
||||||
nlp.add_pipe(DependencyParser(nlp.vocab), name='parser')
|
nlp.add_pipe(DependencyParser(nlp.vocab), name='parser')
|
||||||
nlp.parser.add_label('nsubj')
|
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')
|
doc = nlp.make_doc(u'Australia is a country')
|
||||||
nlp.parser(doc, beam_width=2)
|
nlp.parser(doc, beam_width=2)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user