dataloader:
  dataset:
    _target_: dataloaders.era5.Era5Forecast
    path: data/era5_240/full/
    lead_time_hours: 24
    input_norm_scheme: pangu
    output_norm_scheme: delta24
    data_augmentation: false
    train_split: recent2
    include_vertical_wind_component: true
    load_prev: true
module:
  name: pg4-baseline
  project: atmo-comp
  module:
    _target_: lightning_modules.forecast.ForecastModuleWithCond
    cond_dim: 256
    lr: 1.0e-05
    betas:
    - 0.95
    - 0.99
    weight_decay: 0.05
    num_warmup_steps: 500
    num_training_steps: 20000
    use_graphcast_coeffs: true
    use_prev: true
  backbone:
    _target_: backbones.archesweather.ArchesWeatherCond
    lon_resolution: 240
    lat_resolution: 120
    two_poles: false
    emb_dim: 192
    patch_size:
    - 2
    - 2
    - 2
    window_size:
    - 1
    - 6
    - 10
    use_skip: true
    conv_head: true
    position_embs_dim: 0
    depth_multiplier: 2
    surface_ch: 11
    level_ch: 12
    cond_dim: 256
    mlp_vert: false
    droppath_coeff: 0.2
    dropout: 0
    first_interaction_layer: linear
    axis_attn: true
    n_level_variables: 6
log: true