Minor adjusts

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

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/tests/manufactured-finetune_prithvi_swin_B.yaml b/tests/manufactured-finetune_prithvi_swin_B.yaml
@@ -9,7 +9,7 @@ trainer:
   logger:
     class_path: TensorBoardLogger
     init_args:
-      save_dir: /tmp
+      save_dir: tests/
       name: all_ecos_random
   callbacks:
     - class_path: RichProgressBar
@@ -24,7 +24,7 @@ trainer:
   check_val_every_n_epoch: 1
   log_every_n_steps: 20
   enable_checkpointing: true
-  default_root_dir: /tmp
+  default_root_dir: tests/
 data:
   class_path: GenericNonGeoPixelwiseRegressionDataModule
   init_args:

diff --git a/tests/manufactured-finetune_prithvi_swin_L.yaml b/tests/manufactured-finetune_prithvi_swin_L.yaml
@@ -9,7 +9,7 @@ trainer:
   logger:
     class_path: TensorBoardLogger
     init_args:
-      save_dir: /tmp
+      save_dir: tests/
       name: all_ecos_random
   callbacks:
     - class_path: RichProgressBar
@@ -24,7 +24,7 @@ trainer:
   check_val_every_n_epoch: 1
   log_every_n_steps: 20
   enable_checkpointing: true
-  default_root_dir: /tmp
+  default_root_dir: tests/
 data:
   class_path: GenericNonGeoPixelwiseRegressionDataModule
   init_args:

diff --git a/tests/manufactured-finetune_prithvi_vit_100.yaml b/tests/manufactured-finetune_prithvi_vit_100.yaml
@@ -9,7 +9,7 @@ trainer:
   logger:
     class_path: TensorBoardLogger
     init_args:
-      save_dir: /tmp
+      save_dir: tests/
       name: all_ecos_random
   callbacks:
     - class_path: RichProgressBar
@@ -24,7 +24,7 @@ trainer:
   check_val_every_n_epoch: 1
   log_every_n_steps: 20
   enable_checkpointing: true
-  default_root_dir: /tmp
+  default_root_dir: tests/
 data:
   class_path: GenericNonGeoPixelwiseRegressionDataModule
   init_args:

diff --git a/tests/manufactured-finetune_prithvi_vit_300.yaml b/tests/manufactured-finetune_prithvi_vit_300.yaml
@@ -9,7 +9,7 @@ trainer:
   logger:
     class_path: TensorBoardLogger
     init_args:
-      save_dir: /tmp
+      save_dir: tests/
       name: all_ecos_random
   callbacks:
     - class_path: RichProgressBar
@@ -24,7 +24,7 @@ trainer:
   check_val_every_n_epoch: 1
   log_every_n_steps: 20
   enable_checkpointing: true
-  default_root_dir: /tmp
+  default_root_dir: tests/
 data:
   class_path: GenericNonGeoPixelwiseRegressionDataModule
   init_args: