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

fix typo

Browse Source
This commit is contained in:
Morvan Zhou
2018-06-14 22:20:51 +08:00
parent 921b69a582
commit 79380fb90e
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

@ -82,7 +82,7 @@ for step in range(100):
# !! next step is important !! # !! next step is important !!
h_state = h_state.data # repack the hidden state, break the connection from last iteration h_state = h_state.data # repack the hidden state, break the connection from last iteration
loss = loss_func(prediction, y) # cross entropy loss loss = loss_func(prediction, y) # calculate loss
optimizer.zero_grad() # clear gradients for this training step optimizer.zero_grad() # clear gradients for this training step
loss.backward() # backpropagation, compute gradients loss.backward() # backpropagation, compute gradients
optimizer.step() # apply gradients optimizer.step() # apply gradients