diff --git a/tutorial-contents/403_RNN_regressor.py b/tutorial-contents/403_RNN_regressor.py index c8b19ca..4a959a6 100644 --- a/tutorial-contents/403_RNN_regressor.py +++ b/tutorial-contents/403_RNN_regressor.py @@ -55,6 +55,7 @@ class RNN(nn.Module): # instead, for simplicity, you can replace above codes by follows # r_out = r_out.view(-1, 32) # outs = self.out(r_out) + # outs = outs.view(-1, TIME_STEP, 1) # return outs, h_state rnn = RNN()