explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-06-05 21:53:05 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add option to use gpu to spacy train

Browse Source
This commit is contained in:
Matthew Honnibal 2017-05-18 04:21:49 -05:00
parent a1d8e420b5
commit 39ea38c4b1
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spacy/__main__.py
View File

@ -83,18 +83,20 @@ class CLI(object):
n_iter=("number of iterations", "option", "n", int), n_iter=("number of iterations", "option", "n", int),
nsents=("number of sentences", "option", None, int), nsents=("number of sentences", "option", None, int),
parser_L1=("L1 regularization penalty for parser", "option", "L", float), parser_L1=("L1 regularization penalty for parser", "option", "L", float),
use_gpu=("Use GPU", "flag", "g", bool),
no_tagger=("Don't train tagger", "flag", "T", bool), no_tagger=("Don't train tagger", "flag", "T", bool),
no_parser=("Don't train parser", "flag", "P", bool), no_parser=("Don't train parser", "flag", "P", bool),
no_ner=("Don't train NER", "flag", "N", bool) no_ner=("Don't train NER", "flag", "N", bool)
) )
def train(self, lang, output_dir, train_data, dev_data=None, n_iter=15, def train(self, lang, output_dir, train_data, dev_data=None, n_iter=15,
nsents=0, parser_L1=0.0, no_tagger=False, no_parser=False, no_ner=False): nsents=0, parser_L1=0.0, use_gpu=False,
no_tagger=False, no_parser=False, no_ner=False):
""" """
Train a model. Expects data in spaCy's JSON format. Train a model. Expects data in spaCy's JSON format.
""" """
nsents = nsents or None nsents = nsents or None
cli_train(lang, output_dir, train_data, dev_data, n_iter, nsents, not no_tagger, cli_train(lang, output_dir, train_data, dev_data, n_iter, nsents,
not no_parser, not no_ner, parser_L1) use_gpu, not no_tagger, not no_parser, not no_ner, parser_L1)
@plac.annotations( @plac.annotations(
lang=("model language", "positional", None, str), lang=("model language", "positional", None, str),