refactor: 🎨 Move validation to the base configuration

Addresses part of #7

Author: adjavon

{{cookiecutter.project_slug}}/config/config.yaml

@@ -18,4 +18,23 @@ gt:
 
 
 train: 
-  experiment_dir: ??? # TODO specify the path to save the run results, will also be used to store checkpoints etc.
+  experiment_dir: ??? # TODO specify the path to save the run results, will also be used to store checkpoints etc.
+  input_shape: # TODO specify the input shape of the model
+    - ???
+    - ???
+    - ???
+
+validate: 
+  batch_size: ???
+  container: ${data.container}
+  dataset: ${data.dataset}
+  num_transmitters: ${gt.num_transmitters}
+  experiment_dir: ${train.experiment_dir}
+  checkpoint_number: ???
+  input_shape: ${train.input_shape}
+  num_workers: 12
+  nt_name: ${gt.nt_name}
+  point_id: ${gt.point_id}
+  num_partitions: 10
+  partition_id: 1
+  output_dir: ??? # TODO specify the path to save the validation results, optional

{{cookiecutter.project_slug}}/scripts/03_inference.py

@@ -253,7 +253,7 @@ def validate(
         save_output(results, identifiers, class_names_ordered, output_file)
         logging.info(f"Intermediate results saved to {output_file}")
 
-@hydra.main(config_path="../config", config_name="validate")
+@hydra.main(config_path="../config", config_name="config")
 def main(cfg: DictConfig):
     logging.basicConfig(
         level=logging.INFO,
@@ -267,25 +267,12 @@ def main(cfg: DictConfig):
     # training data
     validation_data = cfg.gt.val
 
-    num_partitions = 1
-    partition_id = 1
-    if "num_partitions" in cfg:
-        num_partitions = cfg.num_partitions
-    if "partition_id" in cfg:
-        partition_id = cfg.partition_id
-    output_dir = None
-    if "output_dir" in cfg:
-        output_dir = cfg.output_dir
-
     # run validation
     validate(
         model=model,
         # Get the data location
         val_gt_location=validation_data,
         **cfg.validate,
-        num_partitions=num_partitions,
-        partition_id=partition_id,
-        output_dir=output_dir,
     )