⚙️ Add testing params for ProcessDict

valentingol · Dec 17, 2023 · 21d0fef · 21d0fef
1 parent 9fb078c
commit 21d0fef
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/tests/configs/integration/test2/default.yaml b/tests/configs/integration/test2/default.yaml
@@ -7,7 +7,6 @@ models:
   vit_b16_cfg_path@merge_add: tests/configs/integration/test2/models/vit_b16.yaml
 
 ---
-
 data:
   dataset_path@type:str: ../data
   dataset_cfg_path@merge_add: tests/configs/integration/test2/data.yaml
@@ -17,3 +16,4 @@ train:
     type@optim_type: SGD
     lr@type:float: 0.01
     momentum@type:float|int: 0.9
+sweep_cfg: None
diff --git a/tests/configs/integration/test2/exp1.yaml b/tests/configs/integration/test2/exp1.yaml
@@ -6,3 +6,8 @@ models.vit_b16:
   in_size: !copy@type:int data.data_size
   dim: 512
 train.n_epochs: 30
+sweep_cfg@dict:
+  method: grid
+  models.dim:
+    - 512
+    - 1024
diff --git a/tests/configs/integration/test2/exp2.yaml b/tests/configs/integration/test2/exp2.yaml
@@ -2,3 +2,8 @@ data.data_size: 512
 data.dataset_path: ../../mydata
 data.augmentation: [RandomHorizontalFlip, RandomVerticalFlip]
 1@merge_add: !delete tests/configs/integration/test2/exp3.yaml
+sweep_cfg@dict:
+  method: bayes
+  data.data_size:
+    - 512
+    - 1024