File size: 442 Bytes
e8d9eff b933866 e8d9eff b933866 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
wandb_mode: "disabled"
vartype: "uvtp122"
train_data_path: "."
valid_data_path: "."
singular_sharded_checkpoint: "magnet-flux-uvtp122-epoch-99-loss-0.1022.pt"
file_glob_pattern: "wxc_input_u_v_t_p_output_theta_uw_vw_era5_*.nc"
lr: 0.0001
hidden_channels: 160
n_lats_px: 64
n_lons_px: 128
in_channels_static: 3
mask_unit_size_px:
- 8
- 16
patch_size_px:
- 1
- 1
### Training Params
max_epochs: 100
batch_size: 12
num_data_workers: 8 |