explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-11-04 01:48:04 +03:00
Code Issues Packages Projects Releases Wiki Activity

Don't use gpu

Browse Source
This commit is contained in:
Matthew Honnibal 2017-05-08 08:39:59 -05:00
parent bef89ef23d
commit 5dffb85184
1 changed files with 1 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
bin/parser/train_ud.py
View File

@ -21,12 +21,6 @@ from thinc.neural import Model
from spacy.es import Spanish from spacy.es import Spanish
from spacy.attrs import POS from spacy.attrs import POS
try:
import cupy
print("Using GPU")
Model.ops = CupyOps()
except ImportError:
pass
from thinc.neural import Model from thinc.neural import Model
@ -103,8 +97,6 @@ def organize_data(vocab, train_sents):
def main(lang_name, train_loc, dev_loc, model_dir, clusters_loc=None): def main(lang_name, train_loc, dev_loc, model_dir, clusters_loc=None):
if cupy is not None:
Model.ops = CupyOps()
LangClass = spacy.util.get_lang_class(lang_name) LangClass = spacy.util.get_lang_class(lang_name)
train_sents = list(read_conllx(train_loc)) train_sents = list(read_conllx(train_loc))
dev_sents = list(read_conllx(dev_loc)) dev_sents = list(read_conllx(dev_loc))
@ -184,10 +176,9 @@ def main(lang_name, train_loc, dev_loc, model_dir, clusters_loc=None):
tokvecs, upd_tokvecs = encoder.begin_update(docs) tokvecs, upd_tokvecs = encoder.begin_update(docs)
for doc, tokvec in zip(docs, tokvecs): for doc, tokvec in zip(docs, tokvecs):
doc.tensor = tokvec doc.tensor = tokvec
d_tokvecs, loss = parser.update(docs, golds, sgd=optimizer) d_tokvecs = parser.update(docs, golds, sgd=optimizer)
upd_tokvecs(d_tokvecs, sgd=optimizer) upd_tokvecs(d_tokvecs, sgd=optimizer)
encoder.update(docs, golds, sgd=optimizer) encoder.update(docs, golds, sgd=optimizer)
nn_loss[-1] += loss
nlp = LangClass(vocab=vocab, parser=parser) nlp = LangClass(vocab=vocab, parser=parser)
scorer = score_model(vocab, encoder, parser, read_conllx(dev_loc)) scorer = score_model(vocab, encoder, parser, read_conllx(dev_loc))
print('%d:\t%.3f\t%.3f\t%.3f' % (itn, scorer.uas, scorer.las, scorer.tags_acc)) print('%d:\t%.3f\t%.3f\t%.3f' % (itn, scorer.uas, scorer.las, scorer.tags_acc))