From 651e901806e3e33746780e89bd015693a33c7b87 Mon Sep 17 00:00:00 2001 From: yunjey Date: Mon, 13 Mar 2017 19:16:19 +0900 Subject: [PATCH] add code to forward prop before loss calculation --- tutorials/00 - PyTorch Basics/main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tutorials/00 - PyTorch Basics/main.py b/tutorials/00 - PyTorch Basics/main.py index 3f0186a..1bd907b 100644 --- a/tutorials/00 - PyTorch Basics/main.py +++ b/tutorials/00 - PyTorch Basics/main.py @@ -72,6 +72,7 @@ optimizer.step() # linear.bias.data.sub_(0.01 * linear.bias.grad.data) # Print out the loss after optimization. +pred = linear(x) loss = criterion(pred, y) print('loss after 1 step optimization: ', loss.data[0])