stresearch / gaia

Global AI Accelerator - GAIA: Code for training and running climate surrogate models
https://stresearch.github.io/gaia/
4 stars 0 forks source link

Test batch norm, dropout, batch size, l2 weight #55

Open ktrapeznikov opened 2 years ago

ktrapeznikov commented 2 years ago

(you can use a subsampled dataset to speed up experiment subsample by 8)

Ablation study, remove or add one element at a time and see how skill changes:

Effect of batch size on performance try scaling it down (but you may have to reduce number of epochs to get number of steps)

Effect of lr?

Generate performance plot for various parameter values