| { |
| "model_name": "Sonny", |
| "sonny_version": "1.0", |
| "architecture_name": "Sonny", |
| "architecture_family": "StepNets", |
| "class_qualname": "networks.sonny.Sonny", |
| "source_checkpoint": "stormbreaker5_checkpoint_ema_epoch_050.pth", |
| "source_epoch": 50, |
| "init_args": { |
| "variables": [ |
| "2m_temperature", |
| "10m_u_component_of_wind", |
| "10m_v_component_of_wind", |
| "mean_sea_level_pressure", |
| "geopotential_50", |
| "geopotential_100", |
| "geopotential_150", |
| "geopotential_200", |
| "geopotential_250", |
| "geopotential_300", |
| "geopotential_400", |
| "geopotential_500", |
| "geopotential_600", |
| "geopotential_700", |
| "geopotential_850", |
| "geopotential_925", |
| "geopotential_1000", |
| "u_component_of_wind_50", |
| "u_component_of_wind_100", |
| "u_component_of_wind_150", |
| "u_component_of_wind_200", |
| "u_component_of_wind_250", |
| "u_component_of_wind_300", |
| "u_component_of_wind_400", |
| "u_component_of_wind_500", |
| "u_component_of_wind_600", |
| "u_component_of_wind_700", |
| "u_component_of_wind_850", |
| "u_component_of_wind_925", |
| "u_component_of_wind_1000", |
| "v_component_of_wind_50", |
| "v_component_of_wind_100", |
| "v_component_of_wind_150", |
| "v_component_of_wind_200", |
| "v_component_of_wind_250", |
| "v_component_of_wind_300", |
| "v_component_of_wind_400", |
| "v_component_of_wind_500", |
| "v_component_of_wind_600", |
| "v_component_of_wind_700", |
| "v_component_of_wind_850", |
| "v_component_of_wind_925", |
| "v_component_of_wind_1000", |
| "temperature_50", |
| "temperature_100", |
| "temperature_150", |
| "temperature_200", |
| "temperature_250", |
| "temperature_300", |
| "temperature_400", |
| "temperature_500", |
| "temperature_600", |
| "temperature_700", |
| "temperature_850", |
| "temperature_925", |
| "temperature_1000", |
| "specific_humidity_50", |
| "specific_humidity_100", |
| "specific_humidity_150", |
| "specific_humidity_200", |
| "specific_humidity_250", |
| "specific_humidity_300", |
| "specific_humidity_400", |
| "specific_humidity_500", |
| "specific_humidity_600", |
| "specific_humidity_700", |
| "specific_humidity_850", |
| "specific_humidity_925", |
| "specific_humidity_1000" |
| ], |
| "in_img_size": [ |
| 120, |
| 240 |
| ], |
| "patch_size": 2, |
| "use_cnn_head": true, |
| "cnn_head_channels": 256, |
| "hidden_size": 384, |
| "depth": 12, |
| "num_heads": 6, |
| "mlp_ratio": 4.0, |
| "step_ratio": 0.5 |
| } |
| } |
|
|
