mirror of https://github.com/explosion/spaCy.git
Update docs for train CLI --use_gpu option (#4927)
This commit is contained in:
parent
708a4d27eb
commit
7ad000fce7
|
@ -372,7 +372,7 @@ $ python -m spacy train [lang] [output_path] [train_path] [dev_path]
|
||||||
| `--n-iter`, `-n` | option | Number of iterations (default: `30`). |
|
| `--n-iter`, `-n` | option | Number of iterations (default: `30`). |
|
||||||
| `--n-early-stopping`, `-ne` | option | Maximum number of training epochs without dev accuracy improvement. |
|
| `--n-early-stopping`, `-ne` | option | Maximum number of training epochs without dev accuracy improvement. |
|
||||||
| `--n-examples`, `-ns` | option | Number of examples to use (defaults to `0` for all examples). |
|
| `--n-examples`, `-ns` | option | Number of examples to use (defaults to `0` for all examples). |
|
||||||
| `--use-gpu`, `-g` | option | Whether to use GPU. Can be either `0`, `1` or `-1`. |
|
| `--use-gpu`, `-g` | option | GPU ID or `-1` for CPU only (default: `-1`). |
|
||||||
| `--version`, `-V` | option | Model version. Will be written out to the model's `meta.json` after training. |
|
| `--version`, `-V` | option | Model version. Will be written out to the model's `meta.json` after training. |
|
||||||
| `--meta-path`, `-m` <Tag variant="new">2</Tag> | option | Optional path to model [`meta.json`](/usage/training#models-generating). All relevant properties like `lang`, `pipeline` and `spacy_version` will be overwritten. |
|
| `--meta-path`, `-m` <Tag variant="new">2</Tag> | option | Optional path to model [`meta.json`](/usage/training#models-generating). All relevant properties like `lang`, `pipeline` and `spacy_version` will be overwritten. |
|
||||||
| `--init-tok2vec`, `-t2v` <Tag variant="new">2.1</Tag> | option | Path to pretrained weights for the token-to-vector parts of the models. See `spacy pretrain`. Experimental. |
|
| `--init-tok2vec`, `-t2v` <Tag variant="new">2.1</Tag> | option | Path to pretrained weights for the token-to-vector parts of the models. See `spacy pretrain`. Experimental. |
|
||||||
|
|
Loading…
Reference in New Issue