Less workers

Joao-L-S-Almeida · Joao-L-S-Almeida · commit f08c2ebbc14c · 2024-07-02T11:37:50.000-03:00
Signed-off-by: João Lucas de Sousa Almeida &lt;joao.l.sa.9.3@gmail.com&gt;
diff --git a/tests/manufactured-finetune_prithvi_swin_B.yaml b/tests/manufactured-finetune_prithvi_swin_B.yaml
@@ -29,7 +29,7 @@ data:
   class_path: GenericNonGeoPixelwiseRegressionDataModule
   init_args:
     batch_size: 2
-    num_workers: 8
+    num_workers: 4
     train_transform:
       - class_path: albumentations.HorizontalFlip
         init_args:
diff --git a/tests/manufactured-finetune_prithvi_swin_L.yaml b/tests/manufactured-finetune_prithvi_swin_L.yaml
@@ -29,7 +29,7 @@ data:
   class_path: GenericNonGeoPixelwiseRegressionDataModule
   init_args:
     batch_size: 2
-    num_workers: 8
+    num_workers: 4
     train_transform:
       - class_path: albumentations.HorizontalFlip
         init_args:
diff --git a/tests/manufactured-finetune_prithvi_vit_100.yaml b/tests/manufactured-finetune_prithvi_vit_100.yaml
@@ -29,7 +29,7 @@ data:
   class_path: GenericNonGeoPixelwiseRegressionDataModule
   init_args:
     batch_size: 2
-    num_workers: 8
+    num_workers: 4
     train_transform:
       - class_path: albumentations.HorizontalFlip
         init_args:
diff --git a/tests/manufactured-finetune_prithvi_vit_300.yaml b/tests/manufactured-finetune_prithvi_vit_300.yaml
@@ -29,7 +29,7 @@ data:
   class_path: GenericNonGeoPixelwiseRegressionDataModule
   init_args:
     batch_size: 2
-    num_workers: 8
+    num_workers: 4
     train_transform:
       - class_path: albumentations.HorizontalFlip
         init_args:
