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

Merge pull request #129 from mariuszrokita/patch-1

Browse Source 
Update main.py
This commit is contained in:
Yunjey Choi
2020-01-28 00:05:19 +09:00
committed by GitHub
parent 32cf716575 ff803b749f
commit 06e04389bd
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

@ -98,7 +98,7 @@ z = y.numpy()
# ================================================================== #
# 4. Input pipline #
# 4. Input pipeline #
# ================================================================== #
# Download and construct CIFAR-10 dataset.
@ -130,7 +130,7 @@ for images, labels in train_loader:
# ================================================================== #
# 5. Input pipline for custom dataset #
# 5. Input pipeline for custom dataset #
# ================================================================== #
# You should build your custom dataset as below.
@ -186,4 +186,4 @@ model = torch.load('model.ckpt')
# Save and load only the model parameters (recommended).
torch.save(resnet.state_dict(), 'params.ckpt')
resnet.load_state_dict(torch.load('params.ckpt'))
resnet.load_state_dict(torch.load('params.ckpt'))