algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-05 16:36:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

add code to forward prop before loss calculation

Browse Source
This commit is contained in:
yunjey
2017-03-13 19:16:19 +09:00
parent a500ce7396
commit 651e901806
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tutorials/00 - PyTorch Basics/main.py
View File

@ -72,6 +72,7 @@ optimizer.step()
# linear.bias.data.sub_(0.01 * linear.bias.grad.data) # linear.bias.data.sub_(0.01 * linear.bias.grad.data)
# Print out the loss after optimization. # Print out the loss after optimization.
pred = linear(x)
loss = criterion(pred, y) loss = criterion(pred, y)
print('loss after 1 step optimization: ', loss.data[0]) print('loss after 1 step optimization: ', loss.data[0])