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