Less workers

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Jul 2, 2024 · f08c2eb · f08c2eb
1 parent 4d18145
commit f08c2eb
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
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: