You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Remove reference to non-existing --model argument (pytorch#1110)
Remove reference to non-existing --model argument
`generate.py` does not accept a `--model` argument, it just uses the default checkpoint (unless you specify `--checkpoint` argument).
# Train a Transformer model on Wikitext-2 with CUDA.
12
12
13
-
python generate.py # Generate samples from the trained LSTM model.
14
-
python generate.py --cuda --model Transformer
15
-
# Generate samples from the trained Transformer model.
13
+
python generate.py # Generate samples from the default model checkpoint.
16
14
```
17
15
18
16
The model uses the `nn.RNN` module (and its sister modules `nn.GRU` and `nn.LSTM`) or Transformer module (`nn.TransformerEncoder` and `nn.TransformerEncoderLayer`) which will automatically use the cuDNN backend if run on CUDA with cuDNN installed.
0 commit comments