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

Add flag to toggle GPU to DyNet code

Browse Source
This commit is contained in:
Matthew Honnibal 2016-11-16 05:51:00 -06:00
parent 3a31c3a961
commit 8f053fd943
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
examples/spacy_dynet_lstm.py
View File

@ -7,14 +7,17 @@ import plac
import random
from collections import Counter
import numpy as np
import os
from collections import defaultdict
from itertools import count
#import _gdynet as dynet
#from _gdynet import cg
import dynet
from dynet import cg
if os.environ.get('DYNET_GPU') == '1':
import _gdynet as dynet
from _gdynet import cg
else:
import dynet
from dynet import cg
class Vocab:
@ -180,7 +183,7 @@ def main(train_loc, dev_loc, model_dir):
tagged = loss = 0
for ITER in xrange(50):
for ITER in xrange(1):
random.shuffle(train)
for i, s in enumerate(train,1):
if i % 5000 == 0: