| 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 |