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

Fix GPU selection in spacy train

Browse Source
This commit is contained in:
Matthw Honnibal 2019-10-18 17:23:55 +02:00
parent 49c0adc706
commit f3e2aaea1e
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/cli/train.py
View File

@ -11,6 +11,7 @@ import srsly
from wasabi import Printer from wasabi import Printer
import contextlib import contextlib
import random import random
from thinc.neural.util import require_gpu
from .._ml import create_default_optimizer from .._ml import create_default_optimizer
from ..attrs import PROB, IS_OOV, CLUSTER, LANG from ..attrs import PROB, IS_OOV, CLUSTER, LANG
@ -85,6 +86,8 @@ def train(
JSON format. To convert data from other formats, use the `spacy convert` JSON format. To convert data from other formats, use the `spacy convert`
command. command.
""" """
if use_gpu != -1:
require_gpu(use_gpu)
# temp fix to avoid import issues cf https://github.com/explosion/spaCy/issues/4200 # temp fix to avoid import issues cf https://github.com/explosion/spaCy/issues/4200
import tqdm import tqdm