From 2c1597a1920085e6684afb39c70a331cc0a965f4 Mon Sep 17 00:00:00 2001 From: subhadarship <35211168+subhadarship@users.noreply.github.com> Date: Thu, 28 Nov 2019 22:44:29 -0500 Subject: [PATCH] more complete description for ArgumentParser --- word_language_model/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/word_language_model/main.py b/word_language_model/main.py index 03bfaeb87e..c60651659d 100644 --- a/word_language_model/main.py +++ b/word_language_model/main.py @@ -10,7 +10,7 @@ import data import model -parser = argparse.ArgumentParser(description='PyTorch Wikitext-2 RNN/LSTM Language Model') +parser = argparse.ArgumentParser(description='PyTorch Wikitext-2 RNN/LSTM/GRU/Transformer Language Model') parser.add_argument('--data', type=str, default='./data/wikitext-2', help='location of the data corpus') parser.add_argument('--model', type=str, default='LSTM',