Description
Adds a config file stretched_grid.yaml intended to be the default when training stretched grid models.
Intended to be used with a cutout dataset, and uses the StretchedTriNodes processor mesh with increased resolution over the limited area.
Processor resolution 5 over the global and 8 over the limited area.
12 connections per processor node in the encoder and 3 conndections per input grid node in the decoder.
Intended to be used with o96 global resolution and 10km limited area resolution for the input grid
Based on the setup used in https://arxiv.org/abs/2409.02891, but with lower resolution. Can be used with e.g. n320 + 2.5km by changing processor resolution to 7+10.
Sample graph using o96 and meps resampled to 10km is attached.
Description Adds a config file stretched_grid.yaml intended to be the default when training stretched grid models.
Based on the setup used in https://arxiv.org/abs/2409.02891, but with lower resolution. Can be used with e.g. n320 + 2.5km by changing processor resolution to 7+10.
Sample graph using o96 and meps resampled to 10km is attached.
stretched_grid.zip