We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 355ffd0 commit 5551061Copy full SHA for 5551061
word_language_model/main.py
@@ -10,7 +10,7 @@
10
import data
11
import model
12
13
-parser = argparse.ArgumentParser(description='PyTorch Wikitext-2 RNN/LSTM Language Model')
+parser = argparse.ArgumentParser(description='PyTorch Wikitext-2 RNN/LSTM/GRU/Transformer Language Model')
14
parser.add_argument('--data', type=str, default='./data/wikitext-2',
15
help='location of the data corpus')
16
parser.add_argument('--model', type=str, default='LSTM',
0 commit comments