Update data_utils.py

data_utils.py

@@ -44,15 +44,10 @@ class HFDatasetWrapper(Dataset):
 def get_train_transform():
     return transforms.Compose(
         [
-            transforms.RandomResizedCrop(IMAGE_SIZE, scale=(0.65, 1.0)),
+            transforms.Resize((IMAGE_SIZE, IMAGE_SIZE)),
             transforms.RandomHorizontalFlip(p=0.5),
-            transforms.RandomVerticalFlip(p=0.2),
-            transforms.RandomRotation(degrees=15),
-            transforms.ColorJitter(
-                brightness=0.2,
-                contrast=0.2,
-                saturation=0.1,
-            ),
+            transforms.RandomVerticalFlip(p=0.5),
+            transforms.RandomRotation(degrees=5),
             transforms.ToTensor(),
             transforms.Normalize(
                 mean=(0.485, 0.456, 0.406),