Switch logging from TensorBoard to Aim and update logging configurations

- Updated configs/logging/test_tensorboard.yaml to include timestamp in version
- Changed default logging in run.yaml from TensorBoard to Aim
- Updated configs/test.yaml to use test_aim logging configuration
- Added aim.yaml and test_aim.yaml to configs/logging

configs/logging/aim.yaml

@@ -0,0 +1,8 @@
+log_every_n_steps: 100
+
+logger:
+  _target_: aim.pytorch_lightning.AimLogger
+  experiment: 'trained_on_${sample_range_trained_on}/${resolution_used}_hnet_model_DOA${max_len}_${now:%H%M%S}'
+  train_metric_prefix: 'train_'
+  test_metric_prefix: 'test_'
+  val_metric_prefix: 'validation_'

configs/logging/test_aim.yaml

@@ -0,0 +1,8 @@
+log_every_n_steps: 100
+
+logger:
+  _target_: aim.pytorch_lightning.AimLogger
+  experiment: 'tested_on_${sample_range_tested_on}/trained_on_${sample_range_trained_on}/${resolution_used}_hnet_model_DOA${max_len}_${now:%H%M%S}'
+  train_metric_prefix: 'train_'
+  test_metric_prefix: 'test_'
+  val_metric_prefix: 'validation_'

configs/logging/test_tensorboard.yaml

@@ -4,6 +4,6 @@ logger:
   _target_: lightning.pytorch.loggers.tensorboard.TensorBoardLogger
   save_dir: "${hydra:runtime.cwd}/tb_logs/${now:%y%m%d}/"
   name: 'hnet_model/tested_on_${sample_range_tested_on}/trained_on_${sample_range_trained_on}/${resolution_used}'
-  version: "hnet_model_DOA${max_len}"
+  version: "hnet_model_DOA${max_len}_${now:%H%M%S}"
   log_graph: False
   default_hp_metric: False

configs/run.yaml

@@ -29,7 +29,7 @@ defaults:
   - callbacks: # Dict of callbacks
       - hnet_checkpoint
       - rich_model_summary
-  - logging: tensorboard
+  - logging: aim
   - metrics: 
     - f1
     - precision

configs/test.yaml

@@ -27,7 +27,7 @@ defaults:
   - trainer: ddp
   - callbacks: # Dict of callbacks
       - rich_model_summary
-  - logging: test_tensorboard
+  - logging: test_aim
   - metrics: f1
   - optimizer: adam
   - _self_ # priority is given to run.yaml for overrides