Skip to content

Commit 5551061

Browse files
authored
more complete description for ArgumentParser (#670)
1 parent 355ffd0 commit 5551061

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

word_language_model/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
import data
1111
import model
1212

13-
parser = argparse.ArgumentParser(description='PyTorch Wikitext-2 RNN/LSTM Language Model')
13+
parser = argparse.ArgumentParser(description='PyTorch Wikitext-2 RNN/LSTM/GRU/Transformer Language Model')
1414
parser.add_argument('--data', type=str, default='./data/wikitext-2',
1515
help='location of the data corpus')
1616
parser.add_argument('--model', type=str, default='LSTM',

0 commit comments

Comments
 (0)