romeokienzler commited on
Commit
ee4c8f1
1 Parent(s): 0aa54c1

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +8 -6
config.yaml CHANGED
@@ -1,9 +1,13 @@
1
  params:
 
2
  input_size_time: 2
3
- patch_size_px: [2, 2]
4
- mask_unit_size_px: [30, 32]
 
5
  n_lats_px: 360
6
  n_lons_px: 576
 
 
7
  embed_dim: 2560
8
  n_blocks_encoder: 12
9
  n_blocks_decoder: 2
@@ -12,7 +16,5 @@ params:
12
  dropout: 0.0
13
  drop_path: 0.0
14
  parameter_dropout: 0.0
15
- residual: True # Or False, depending on your needs
16
- masking_mode: 'random' # Or 'fixed'
17
- decoder_shifting: True # Or False
18
- positional_encoding: True # Or False
 
1
  params:
2
+ in_channels: 160
3
  input_size_time: 2
4
+ in_channels_static: 8
5
+ input_scalers_epsilon: 0.0
6
+ static_input_scalers_epsilon: 0.0
7
  n_lats_px: 360
8
  n_lons_px: 576
9
+ patch_size_px: [2, 2]
10
+ mask_unit_size_px: [30, 32]
11
  embed_dim: 2560
12
  n_blocks_encoder: 12
13
  n_blocks_decoder: 2
 
16
  dropout: 0.0
17
  drop_path: 0.0
18
  parameter_dropout: 0.0
19
+ checkpoint_encoder: []
20
+ checkpoint_decoder: []