Merge pull request yunjey#129 from mariuszrokita/patch-1

yunjey · web-flow · commit 06e04389bd3e · 2020-01-28T00:05:19.000+09:00
Update main.py
diff --git a/tutorials/01-basics/pytorch_basics/main.py b/tutorials/01-basics/pytorch_basics/main.py
@@ -98,7 +98,7 @@
 
 
 # ================================================================== #
-#                         4. Input pipline                           #
+#                         4. Input pipeline                           #
 # ================================================================== #
 
 # Download and construct CIFAR-10 dataset.
@@ -130,7 +130,7 @@
 
 
 # ================================================================== #
-#                5. Input pipline for custom dataset                 #
+#                5. Input pipeline for custom dataset                 #
 # ================================================================== #
 
 # You should build your custom dataset as below.
@@ -186,4 +186,4 @@ def __len__(self):
 
 # 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'))
