algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-10 13:49:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #128 from timctho/master

Browse Source 
Fix requires_grad=True needs float tensor
This commit is contained in:
Yunjey Choi
2018-08-07 12:46:30 +09:00
committed by GitHub
parent 9219526869 2bfebb0a4a
commit 6fe5ff5211
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tutorials/01-basics/pytorch_basics/main.py
View File

@ -23,9 +23,9 @@ import torchvision.transforms as transforms
# ================================================================== # # ================================================================== #
# Create tensors. # Create tensors.
x = torch.tensor(1, requires_grad=True) x = torch.tensor(1., requires_grad=True)
w = torch.tensor(2, requires_grad=True) w = torch.tensor(2., requires_grad=True)
b = torch.tensor(3, requires_grad=True) b = torch.tensor(3., requires_grad=True)
# Build a computational graph. # Build a computational graph.
y = w * x + b # y = 2 * x + 3 y = w * x + b # y = 2 * x + 3