explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-25 17:36:30 +03:00
Code Issues Packages Projects Releases Wiki Activity

Copy vectors to GPU on begin training

Browse Source
This commit is contained in:
Matthew Honnibal 2017-09-18 18:04:16 -05:00
parent 3fa76c17d1
commit c858927271
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/language.py
View File

@ -364,6 +364,9 @@ class Language(object):
device.use() device.use()
Model.ops = CupyOps() Model.ops = CupyOps()
Model.Ops = CupyOps Model.Ops = CupyOps
if self.vocab.vectors.data.shape[1] >= 1:
self.vocab.vectors.data = Model.ops.asarray(
self.vocab.vectors.data)
else: else:
device = None device = None
for proc in self.pipeline: for proc in self.pipeline: