From 3e3b0e9d10ce943eb126d9617d0c4f6595c3a49d Mon Sep 17 00:00:00 2001 From: Morvan Zhou Date: Thu, 11 May 2017 16:53:32 +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 ad79500..1d68b86 100644 --- a/tutorial-contents/403_RNN_regressor.py +++ b/tutorial-contents/403_RNN_regressor.py @@ -45,7 +45,7 @@ class RNN(nn.Module): def forward(self, x, h_state): # x (batch, time_step, input_size) # h_state (n_layers, batch, hidden_size) - # r_out (batch, time_step, output_size) + # r_out (batch, time_step, hidden_size) r_out, h_state = self.rnn(x, h_state) outs = [] # save all predictions