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-10 20:18:33 +10:00
parent e2f10e0ce1
commit 3d8df0c297
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tutorial-contents/402_RNN_classifier.py
View File

@ -56,7 +56,7 @@ class RNN(nn.Module):
super(RNN, self).__init__() super(RNN, self).__init__()
self.rnn = nn.LSTM( # if use nn.RNN(), it hardly learns self.rnn = nn.LSTM( # if use nn.RNN(), it hardly learns
input_size=28, input_size=INPUT_SIZE,
hidden_size=64, # rnn hidden unit hidden_size=64, # 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)