File size: 369 Bytes
f812633 6b0a83c f812633 6b0a83c f812633 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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 |