Update data_utils.py

data_utils.py

@@ -37,11 +37,11 @@ class HFDatasetWrapper(Dataset):
 def get_transform():
     return transforms.Compose(
         [
-            transforms.Resize((IMAGE_SIZE, IMAGE_SIZE)),
+            transforms.Resize((224, 224)),
             transforms.ToTensor(),
             transforms.Normalize(
-                mean=(0.5, 0.5, 0.5),
-                std=(0.5, 0.5, 0.5),
+                mean=(0.485, 0.456, 0.406),
+                std=(0.229, 0.224, 0.225),
             ),
         ]
     )