| params: | |
| input_size_time: 2 | |
| patch_size_px: [2, 2] | |
| mask_unit_size_px: [30, 32] | |
| n_lats_px: 360 | |
| n_lons_px: 576 | |
| 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 | |
| residual: True | |
| masking_mode: 'random' | |
| decoder_shifting: True | |
| positional_encoding: True |