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

update

Browse Source
This commit is contained in:
Morvan Zhou
2017-05-11 16:58:23 +10:00
parent 3e3b0e9d10
commit e95d798e68
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tutorial-contents/403_RNN_regressor.py
View File

@ -35,7 +35,7 @@ class RNN(nn.Module):
super(RNN, self).__init__() super(RNN, self).__init__()
self.rnn = nn.RNN( self.rnn = nn.RNN(
input_size=1, input_size=INPUT_SIZE,
hidden_size=32, # rnn hidden unit hidden_size=32, # rnn hidden unit
num_layers=1, # number of rnn layer num_layers=1, # number of rnn layer
batch_first=True, # input & output will has batch size as 1s dimension. e.g. (batch, time_step, input_size) batch_first=True, # input & output will has batch size as 1s dimension. e.g. (batch, time_step, input_size)