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

Merge pull request #140 from litcoderr/master

Browse Source 
simple typo
This commit is contained in:
Yunjey Choi
2018-11-04 18:37:57 +09:00
committed by GitHub
parent 6fe5ff5211 16d6a76e90
commit 6f28678302
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -133,7 +133,7 @@ for images, labels in train_loader:
# 5. Input pipline for custom dataset # # 5. Input pipline for custom dataset #
# ================================================================== # # ================================================================== #
# You should your build your custom dataset as below. # You should build your custom dataset as below.
class CustomDataset(torch.utils.data.Dataset): class CustomDataset(torch.utils.data.Dataset):
def __init__(self): def __init__(self):
# TODO # TODO