Update main.py

Fixed typos.

Author: mariuszrokita

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 your 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'))