Update Task 1.py

yifan1207 · Jun 11, 2024 · 2b355ea · 2b355ea
1 parent 4d2f42d
commit 2b355ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/QMLS Code/Task 1.py b/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