Refactor Callbacks

CHANGELOG.md

@@ -37,6 +37,7 @@ Keep it human-readable, your future self will thank you!
 ### Changed
 
 - Updated configuration examples in documentation and corrected links - [#46](https://github.com/ecmwf/anemoi-training/pull/46)
+- Refactored callbacks. [#60](https://github.com/ecmwf/anemoi-training/pulls/60)
 
 ## [0.1.0 - Anemoi training - First release](https://github.com/ecmwf/anemoi-training/releases/tag/0.1.0) - 2024-08-16
 

src/anemoi/training/config/diagnostics/eval_rollout.yaml

@@ -70,6 +70,12 @@ checkpoint:
     save_frequency: null # Does not scale with rollout
     num_models_saved: 0
 
+callbacks:
+  # Add callbacks here
+
+plot_callbacks:
+  # Add extra plot callbacks here
+
 log:
   wandb:
     enabled: False