explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-19 20:52:23 +03:00
Code Issues Packages Projects Releases Wiki Activity

On initialize, use just two samples

Browse Source 
Coref docs are kind of long, and using 10 samples on a smallish GPU can
cause OOMs.
This commit is contained in:
Paul O'Leary McCann 2021-07-03 18:43:03 +09:00
parent 251a5b43ac
commit 2d3c559dc4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline/coref.py
View File

@ -333,7 +333,7 @@ class CoreferenceResolver(TrainablePipe):
X = [] X = []
Y = [] Y = []
for ex in islice(get_examples(), 10): for ex in islice(get_examples(), 2):
X.append(ex.predicted) X.append(ex.predicted)
Y.append(ex.reference) Y.append(ex.reference)