Skip to content

Commit 0252bda

Browse files
authored
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).
1 parent e4e8da8 commit 0252bda

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

word_language_model/README.md

+1-3
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,7 @@ python main.py --cuda --tied # Train a tied LSTM on Wikitext-2 wit
1010
python main.py --cuda --epochs 6 --model Transformer --lr 5
1111
# Train a Transformer model on Wikitext-2 with CUDA.
1212

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.
1614
```
1715

1816
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

Comments
 (0)