From e95d798e680c852ebb619aa3e44214cc0f4096d6 Mon Sep 17 00:00:00 2001 From: Morvan Zhou Date: Thu, 11 May 2017 16:58:23 +1000 Subject: [PATCH] update --- tutorial-contents/403_RNN_regressor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorial-contents/403_RNN_regressor.py b/tutorial-contents/403_RNN_regressor.py index 1d68b86..2ca3997 100644 --- a/tutorial-contents/403_RNN_regressor.py +++ b/tutorial-contents/403_RNN_regressor.py @@ -35,7 +35,7 @@ class RNN(nn.Module): super(RNN, self).__init__() self.rnn = nn.RNN( - input_size=1, + input_size=INPUT_SIZE, hidden_size=32, # 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)