From 2bfebb0a4a8367d8ea0f68ca44e586a520dbd9e9 Mon Sep 17 00:00:00 2001 From: TimHo Date: Sun, 5 Aug 2018 23:59:17 +0800 Subject: [PATCH] Fix requires_grad=True needs float tensor --- tutorials/01-basics/pytorch_basics/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tutorials/01-basics/pytorch_basics/main.py b/tutorials/01-basics/pytorch_basics/main.py index f958d7c..78b692a 100644 --- a/tutorials/01-basics/pytorch_basics/main.py +++ b/tutorials/01-basics/pytorch_basics/main.py @@ -23,9 +23,9 @@ import torchvision.transforms as transforms # ================================================================== # # Create tensors. -x = torch.tensor(1, requires_grad=True) -w = torch.tensor(2, requires_grad=True) -b = torch.tensor(3, requires_grad=True) +x = torch.tensor(1., requires_grad=True) +w = torch.tensor(2., requires_grad=True) +b = torch.tensor(3., requires_grad=True) # Build a computational graph. y = w * x + b # y = 2 * x + 3