Merge pull request #5 from szmazurek/update_base_configs

Reorganize test configs
mlcommons · Oct 4, 2024 · 4810cdd · 4810cdd
2 parents 9b875d0 + db3fd6e
commit 4810cdd
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 20 deletions.
diff --git a/testing/configs/module_config_dcgan.yaml b/testing/configs/module_config_dcgan.yaml
@@ -1,4 +1,3 @@
-batch_size: 1
 data_augmentation: {}
 data_postprocessing: {}
 data_preprocessing:
@@ -18,6 +17,7 @@ data_preprocessing:
     resize:
     - 64
     - 64
+
 dataloader_config:
   inference:
     drop_last: false
@@ -39,12 +39,8 @@ dataloader_config:
     num_workers: 0
     pin_memory: false
     shuffle: false
-in_memory: false
-inference_parameters:
-  batch_size: 1
-  n_images_to_generate: 10
 
-modality: rad
+
 model_config:
   architecture:
     latent_vector_size: 100
@@ -91,10 +87,13 @@ model_config:
   n_fixed_images_to_generate: 10
   fixed_images_batch_size: 10
 
+inference_parameters:
+  batch_size: 1
+  n_images_to_generate: 10
 num_epochs: 1
-patience: 0
+batch_size: 1
+modality: rad
 save_model_every_n_epochs: 1
-verbose: true
 version:
   maximum: 0.0.20-dev
   minimum: 0.0.14
diff --git a/testing/configs/module_config_ddpm.yaml b/testing/configs/module_config_ddpm.yaml
@@ -1,4 +1,3 @@
-batch_size: 1
 data_augmentation: {}
 data_postprocessing: {}
 data_preprocessing:
@@ -18,6 +17,7 @@ data_preprocessing:
     resize:
     - 64
     - 64
+
 dataloader_config:
   inference:
     drop_last: false
@@ -39,11 +39,8 @@ dataloader_config:
     num_workers: 0
     pin_memory: false
     shuffle: false
-inference_parameters:
-  batch_size: 1
-  n_images_to_generate: 1
-in_memory: false
-modality: rad
+
+
 model_config:
   architecture:
     num_eval_timesteps: 1
@@ -65,10 +62,14 @@ model_config:
   schedulers:
     type: triangle
     step_size: 2
+
+inference_parameters:
+  batch_size: 1
+  n_images_to_generate: 1
+modality: rad
 num_epochs: 1
-patience: 0
+batch_size: 1
 save_model_every_n_epochs: 1
-verbose: true
 version:
   maximum: 0.0.20-dev
   minimum: 0.0.14
diff --git a/testing/configs/module_config_vqvae.yaml b/testing/configs/module_config_vqvae.yaml
@@ -1,4 +1,3 @@
-batch_size: 1
 data_augmentation: {}
 data_postprocessing: {}
 data_preprocessing:
@@ -18,6 +17,7 @@ data_preprocessing:
     resize:
     - 64
     - 64
+
 dataloader_config:
   inference:
     drop_last: false
@@ -39,7 +39,7 @@ dataloader_config:
     num_workers: 0
     pin_memory: false
     shuffle: false
-modality: rad
+
 model_config:
   converter_type: soft
   labeling_paradigm: unlabeled
@@ -62,12 +62,14 @@ model_config:
   tensor_shape:
   - 64
   - 64
+
 inference_parameters:
   batch_size: 1
+modality: rad
+batch_size: 1
 num_epochs: 1
-patience: 0
 save_model_every_n_epochs: 1
-verbose: true
 version:
   maximum: 0.0.20-dev
   minimum: 0.0.14
+