Update Task 1.py

Author: Yifan Zhou

QMLS Code/Task 1.py

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