Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Set cupy.random seed in fix_random_seed helper

This commit is contained in:
Matthew Honnibal 2018-12-08 12:37:38 +01:00
parent ffdd5e964f
commit 11a29af751
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/util.py
View File

@ -15,6 +15,11 @@ import itertools
import numpy.random
import srsly
try:
import cupy.random
except ImportError:
cupy = None
from .symbols import ORTH
from .compat import cupy, CudaStream, path2str, basestring_, unicode_
from .compat import import_file
@ -598,6 +603,8 @@ def use_gpu(gpu_id):
def fix_random_seed(seed=0):
random.seed(seed)
numpy.random.seed(seed)
if cupy is not None:
cupy.random.seed(seed)
class SimpleFrozenDict(dict):