Update Task 1.py

Author: Yifan Zhou

QMLS Code/Task 1.py

@@ -1,6 +1,6 @@
 import torch    
 import torch.nn as nn    
-import torch.optim as optim 
+import torch.optim as optim  
 import torchvision
 import torchvision.transforms as transforms
 import numpy as np