Skip to content

Commit fa54b09

Browse files
committed
minor edits
1 parent 8e64bb2 commit fa54b09

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
# grid-lstm-tensorflow
2+
23
Examples of using GridLSTM (and GridRNN in general) in tensorflow

char-rnn/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def main():
2020
parser.add_argument('--num_layers', type=int, default=2,
2121
help='number of layers in the RNN')
2222
parser.add_argument('--model', type=str, default='lstm',
23-
help='rnn, gru, or lstm')
23+
help='rnn, gru, lstm, gridlstm, gridgru')
2424
parser.add_argument('--batch_size', type=int, default=50,
2525
help='minibatch size')
2626
parser.add_argument('--seq_length', type=int, default=50,

0 commit comments

Comments
 (0)