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

Merge pull request #112 from marlonjan/patch-1

Browse Source 
fix typo in comments
This commit is contained in:
Yunjey Choi
2018-05-10 10:49:47 +09:00
committed by GitHub
parent 1760cc8e81 63db681c4f
commit 1016ac031f
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -12,8 +12,8 @@ from torch.autograd import Variable
# 1. Basic autograd example 1 (Line 21 to 36) # 1. Basic autograd example 1 (Line 21 to 36)
# 2. Basic autograd example 2 (Line 39 to 77) # 2. Basic autograd example 2 (Line 39 to 77)
# 3. Loading data from numpy (Line 80 to 83) # 3. Loading data from numpy (Line 80 to 83)
# 4. Implementing the input pipline (Line 86 to 113) # 4. Implementing the input pipeline (Line 86 to 113)
# 5. Input pipline for custom dataset (Line 116 to 138) # 5. Input pipeline for custom dataset (Line 116 to 138)
# 6. Using pretrained model (Line 141 to 155) # 6. Using pretrained model (Line 141 to 155)
# 7. Save and load model (Line 158 to 165) # 7. Save and load model (Line 158 to 165)
@ -83,7 +83,7 @@ b = torch.from_numpy(a) # convert numpy array to torch tensor
c = b.numpy() # convert torch tensor to numpy array c = b.numpy() # convert torch tensor to numpy array
#===================== Implementing the input pipline =====================# #===================== Implementing the input pipeline =====================#
# Download and construct dataset. # Download and construct dataset.
train_dataset = dsets.CIFAR10(root='../data/', train_dataset = dsets.CIFAR10(root='../data/',
train=True, train=True,
@ -113,7 +113,7 @@ for images, labels in train_loader:
pass pass
#===================== Input pipline for custom dataset =====================# #===================== Input pipeline for custom dataset =====================#
# You should build custom dataset as below. # You should build custom dataset as below.
class CustomDataset(data.Dataset): class CustomDataset(data.Dataset):
def __init__(self): def __init__(self):