explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-07 15:10:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
Richard Liaw 2020-06-22 19:11:26 -07:00
parent c41d0dcbce
commit 14a42cbc4a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/util.py
View File

@ -898,6 +898,7 @@ def escape_html(text):
def use_gpu(gpu_id): def use_gpu(gpu_id):
return require_gpu(gpu_id) return require_gpu(gpu_id)
def gpu_is_available(): def gpu_is_available():
try: try:
cupy.cuda.runtime.getDeviceCount() cupy.cuda.runtime.getDeviceCount()
@ -905,6 +906,7 @@ def gpu_is_available():
except cupy.cuda.runtime.CUDARuntimeError: except cupy.cuda.runtime.CUDARuntimeError:
return False return False
def fix_random_seed(seed=0): def fix_random_seed(seed=0):
random.seed(seed) random.seed(seed)
numpy.random.seed(seed) numpy.random.seed(seed)