andre/PyTorch-Tutorial
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fill up the missing view() in forward() in comment

Browse Source
This commit is contained in:
keineahnung2345
2018-11-12 14:18:49 +08:00
committed by GitHub
parent 40bcdb5c6c
commit 2748b02bd6
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tutorial-contents/403_RNN_regressor.py
View File

@ -55,6 +55,7 @@ class RNN(nn.Module):
# instead, for simplicity, you can replace above codes by follows # instead, for simplicity, you can replace above codes by follows
# r_out = r_out.view(-1, 32) # r_out = r_out.view(-1, 32)
# outs = self.out(r_out) # outs = self.out(r_out)
# outs = outs.view(-1, TIME_STEP, 1)
# return outs, h_state # return outs, h_state
rnn = RNN() rnn = RNN()