From c3b6395891314c27d0d9fc640ef1b99d4d326fcd Mon Sep 17 00:00:00 2001 From: J911 Date: Fri, 18 Jan 2019 02:08:00 +0900 Subject: [PATCH] fix: change the output method --- 05_linear_regression.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/05_linear_regression.py b/05_linear_regression.py index 0afd430..8285504 100644 --- a/05_linear_regression.py +++ b/05_linear_regression.py @@ -41,7 +41,7 @@ def forward(self, x): # Compute and print loss loss = criterion(y_pred, y_data) - print(epoch, loss.data[0]) + print(epoch, loss.item()) # Zero gradients, perform a backward pass, and update the weights. optimizer.zero_grad()