From e7599c0f14ff0cd7c86fdc329d77fadd1edd7342 Mon Sep 17 00:00:00 2001 From: Morvan Zhou Date: Wed, 9 Aug 2017 01:51:16 +1000 Subject: [PATCH] update --- tutorial-contents/403_RNN_regressor.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tutorial-contents/403_RNN_regressor.py b/tutorial-contents/403_RNN_regressor.py index 7a0d018..8563d45 100644 --- a/tutorial-contents/403_RNN_regressor.py +++ b/tutorial-contents/403_RNN_regressor.py @@ -53,6 +53,10 @@ class RNN(nn.Module): outs.append(self.out(r_out[:, time_step, :])) return torch.stack(outs, dim=1), h_state + # instead, for simplicity, you can replace above codes by follows + # r_out = r_out.view(-1, 32) + # outs = self.out(r_out) + # return outs, h_state rnn = RNN() print(rnn)