From a5eec06271385e9599441b2210e180cfdfdfa13e Mon Sep 17 00:00:00 2001
From: Yifan Zhou <yifan.zhou11882-bigz@basischina.com>
Date: Tue, 11 Jun 2024 10:56:39 +0800
Subject: [PATCH] Update Task 1.py

---
 QMLS Code/Task 1.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/QMLS Code/Task 1.py b/QMLS Code/Task 1.py
index 03daf90..b3d2613 100644
--- 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