IBM
diff --git a/‎tests/manufactured-finetune_prithvi_swin_B_segmentation.yaml
+1-1 b/‎tests/manufactured-finetune_prithvi_swin_B_segmentation.yaml
+1-1
diff --git a/‎tests/segmentation_test_input_2.tif
1.18 MB b/‎tests/segmentation_test_input_2.tif
1.18 MB
diff --git a/‎tests/segmentation_test_input_3.tif
1.18 MB b/‎tests/segmentation_test_input_3.tif
1.18 MB
diff --git a/‎tests/segmentation_test_label_2.tif
393 KB b/‎tests/segmentation_test_label_2.tif
393 KB
diff --git a/‎tests/segmentation_test_label_3.tif
393 KB b/‎tests/segmentation_test_label_3.tif
393 KB
@@ -28,7 +28,7 @@ trainer:
 data:
   class_path: GenericNonGeoSegmentationDataModule
   init_args:
-    batch_size: 2
+    batch_size: 4
     num_workers: 4
     train_transform:
       - class_path: albumentations.HorizontalFlip