From 3d8df0c297a58295b304f0c74c31a3ffcc8e57c9 Mon Sep 17 00:00:00 2001 From: Morvan Zhou Date: Wed, 10 May 2017 20:18:33 +1000 Subject: [PATCH] update --- tutorial-contents/402_RNN_classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorial-contents/402_RNN_classifier.py b/tutorial-contents/402_RNN_classifier.py index 679a255..1909387 100644 --- a/tutorial-contents/402_RNN_classifier.py +++ b/tutorial-contents/402_RNN_classifier.py @@ -56,7 +56,7 @@ class RNN(nn.Module): super(RNN, self).__init__() self.rnn = nn.LSTM( # if use nn.RNN(), it hardly learns - input_size=28, + input_size=INPUT_SIZE, hidden_size=64, # rnn hidden unit num_layers=1, # number of rnn layer batch_first=True, # input & output will has batch size as 1s dimension. e.g. (batch, time_step, input_size)