Update Task 1.py

Author: Yifan Zhou

QMLS Code/Task 1.py

@@ -1,5 +1,5 @@
 import torch  
-import torch.nn as nn  
+import torch.nn as nn   
 import torch.optim as optim
 import torchvision
 import torchvision.transforms as transforms