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

typo

Browse Source
This commit is contained in:
litcoderr
2018-11-04 11:00:43 +09:00
parent 6fe5ff5211
commit 16d6a76e90
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