From 2748b02bd64343ddbef52c083250b4f045a1abf0 Mon Sep 17 00:00:00 2001 From: keineahnung2345 Date: Mon, 12 Nov 2018 14:18:49 +0800 Subject: [PATCH] fill up the missing view() in forward() in comment --- tutorial-contents/403_RNN_regressor.py | 1 + 1 file changed, 1 insertion(+) 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()