diff --git a/QMLS Code/Task 1.py b/QMLS Code/Task 1.py index a3e3b4e..390bd3d 100644 --- a/QMLS Code/Task 1.py +++ b/QMLS Code/Task 1.py @@ -5,7 +5,6 @@ import torchvision.transforms as transforms import numpy as np import os - # Define data transforms data_transforms = { 'train': transforms.Compose([