Repository navigation
Cleaned-up code to train single-column (M1), multiple columns (M2), and global Attention U-Net (M3) to train for all levels in the atmosphere or only in the stratosphere using resolved ERA5 gravity waves fluxes, and then transfer learn on high-resolution climate model output.
Contributors: @TomMelt, @amangupta2, @omarjamil, Aditi Sheshadri, @j-emberton