algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-06 01:15:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

add examples to pytorch basics

Browse Source
This commit is contained in:
yunjey
2017-03-12 14:06:00 +09:00
parent f1d21a69fa
commit eadb0f9580
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,10 +27,10 @@ b = Variable(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
# Compute gradients # Compute gradients.
y.backward() y.backward()
# Print out the gradients # Print out the gradients.
print(x.grad) # x.grad = 2 print(x.grad) # x.grad = 2
print(w.grad) # w.grad = 1 print(w.grad) # w.grad = 1
print(b.grad) # b.grad = 1 print(b.grad) # b.grad = 1
@ -146,7 +146,7 @@ for param in resnet.parameters():
# Replace top layer for finetuning. # Replace top layer for finetuning.
resnet.fc = nn.Linear(resnet.fc.in_features, 100) # 100 is for example. resnet.fc = nn.Linear(resnet.fc.in_features, 100) # 100 is for example.
# For test # For test.
images = Variable(torch.randn(10, 3, 256, 256)) images = Variable(torch.randn(10, 3, 256, 256))
outputs = resnet(images) outputs = resnet(images)
print (outputs.size()) # (10, 100) print (outputs.size()) # (10, 100)