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

Make pretraining script work without GPU

Browse Source
This commit is contained in:
Matthew Honnibal 2018-11-04 17:09:52 +01:00
parent bc8cda818c
commit 09aa616182
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/training/pretrain_textcat.py
View File

@ -21,7 +21,6 @@ from spacy.util import minibatch, use_gpu, compounding
import tqdm
from spacy._ml import Tok2Vec
from spacy.pipeline import TextCategorizer
import cupy.random
import numpy
@ -57,6 +56,8 @@ def prefer_gpu():
if used is None:
return False
else:
import cupy.random
cupy.random.seed(0)
return True
@ -177,7 +178,6 @@ def evaluate_textcat(tokenizer, textcat, texts, cats):
def main(width: int, embed_size: int, vectors_model,
pretrain_iters=30, train_iters=30, train_examples=1000):
random.seed(0)
cupy.random.seed(0)
numpy.random.seed(0)
use_gpu = prefer_gpu()
print("Using GPU?", use_gpu)