File size: 428 Bytes
0aa54c1 ee4c8f1 0aa54c1 ee4c8f1 0aa54c1 ee4c8f1 9f96158 0aa54c1 9f96158 0aa54c1 ee4c8f1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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: [] |