Update Task 1.py

yifan1207 · Jun 11, 2024 · a5eec06 · a5eec06
1 parent acfae0a
commit a5eec06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/QMLS Code/Task 1.py b/QMLS Code/Task 1.py
@@ -2,7 +2,7 @@
 import torch.nn as nn       
 import torch.optim as optim    
 import torchvision
-import torchvision.transforms as transforms
+import torchvision.transforms as transforms 
 import numpy as np
 import os
 # Define data transforms