search

microsoft / aurora

Implementation of the Aurora model for atmospheric forecasting
https://microsoft.github.io/aurora
Other
247 stars 31 forks source link

Rework ID-based parametrisation #27

Closed wesselb closed 2 months ago

wesselb commented 2 months ago

This should make it easier to extend the model to new variables.

Closes #24

scottcha commented 2 months ago

I checked this PR against my fine tuning and was able to add variables w/out needing to directly modify the layers (all completed on the Small version of Aurora). Adding the ERA5 tsn var w/out any additional fine tuning results in image

After a few mins of fine tuning I get this for tsn image

For the existing 2t var I get this w/out finetuning image

And this after a few mins of fine tuning image

While I'm not super experienced yet with leveraging this model so far the behavior seems to be expected and its possible to add vars more easily than previously.

wesselb commented 2 months ago

Thanks, @scottcha! I'm going to interpret this as evidence that the refactoring works as intended and proceed to merging. :)