From cf16e27b7b0a04a1b62971acb014f2adf80c6c56 Mon Sep 17 00:00:00 2001 From: Sopot Cela <93710482+sopotc@users.noreply.github.com> Date: Fri, 3 Feb 2023 17:12:05 +0000 Subject: [PATCH] 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). --- word_language_model/README.md | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/word_language_model/README.md b/word_language_model/README.md index 18341e84c9..254b726585 100644 --- a/word_language_model/README.md +++ b/word_language_model/README.md @@ -10,9 +10,7 @@ python main.py --cuda --tied # Train a tied LSTM on Wikitext-2 wit python main.py --cuda --epochs 6 --model Transformer --lr 5 # Train a Transformer model on Wikitext-2 with CUDA. -python generate.py # Generate samples from the trained LSTM model. -python generate.py --cuda --model Transformer - # Generate samples from the trained Transformer model. +python generate.py # Generate samples from the default model checkpoint. ``` 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.