| params: | |
| in_channels: 160 | |
| input_size_time: 2 | |
| in_channels_static: 8 | |
| input_scalers_epsilon: 0.0 | |
| static_input_scalers_epsilon: 0.0 | |
| n_lats_px: 360 | |
| n_lons_px: 576 | |
| patch_size_px: [2, 2] | |
| mask_unit_size_px: [30, 32] | |
| embed_dim: 2560 | |
| n_blocks_encoder: 12 | |
| n_blocks_decoder: 2 | |
| mlp_multiplier: 4 | |
| n_heads: 16 | |
| dropout: 0.0 | |
| drop_path: 0.0 | |
| parameter_dropout: 0.0 | |
| checkpoint_encoder: [] | |
| checkpoint_decoder: [] |