Testing coverage for Transfer learning #989
Description
Is your feature request related to a problem? Please describe.
As part of a discussion between @mc4117 @ThomasRieutord @mo-mburnand we identified the following tests could be added to anemoi-training to improve sections related to transfer learning or freezing.
- Currently freeze_submodule_by_name function has no unit tests - @JesperDramsch likely this is something you can address in your PR?
- Transfer learning could be tested as part of the integration tests - where we could resume a model trained on o32 to o48
- In general transfer learning could also have better unit-test coverage to ensure that the function is doing what we expect. Some tests already exist under https://github.com/ecmwf/anemoi-core/blob/main/training/tests/unit/train/test_checkpoint_loading.py.
- Specific uses cases such as AICON could be extended - in particular we could consider one related to CERRA combined with transfer learning (@dietervdb-meteo @dnerini - to provide input on the details of this test)
Mariana, Thomas and Matt feel free to add to this!
Describe the solution you'd like
No response
Describe alternatives you've considered
No response
Additional context
No response
Organisation
No response