model_name: simvp_v3_wsd
seed: 27267831
trainer:
  _target_: lightning.pytorch.trainer.trainer.Trainer
  accelerator: gpu
  devices:
  - 1
  precision: 16-mixed
  gradient_clip_val: 0.5
  detect_anomaly: false
  overfit_batches: 0.0
  min_epochs: null
  max_epochs: 80
  reload_dataloaders_every_n_epochs: 0
  num_sanity_val_steps: 8
  fast_dev_run: false
  accumulate_grad_batches: 1
  limit_train_batches: 12000
  limit_val_batches: 400
  log_every_n_steps: 50
model:
  _target_: sat_pred.training_module.TrainingModule
  model:
    _target_: sat_pred.models.simvp_model.SimVP
    num_channels: 11
    history_len: 12
    forecast_len: 12
    hid_S: 64
    hid_T: 512
    N_S: 4
    N_T: 4
  restore_optimizer_state: true
  optimizer:
    _target_: sat_pred.optimizers.AdamWWSD
    lr: 0.0005
    warmup_epochs: 0
    decay_epochs: 30
    total_epochs: 80
    final_factor: 0.02
    weight_decay: 0.01
  target_loss: MAE
  video_plot_t0_times:
  - '2018-06-08T12:00:00'
  - '2018-12-26T12:00:00'
  - '2018-06-14T12:00:00'
  - '2018-05-08T12:00:00'
  - '2018-05-12T12:00:00'
  - '2018-01-29T12:00:00'
  video_crop_plots: null
datamodule:
  _target_: sat_pred.dataset.SatelliteDataModule
  zarr_path:
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2008-3000m-uk-only.zarr
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2009-3000m-uk-only.zarr
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2010-3000m-uk-only.zarr
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2011-3000m-uk-only.zarr
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2012-3000m-uk-only.zarr
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2013-3000m-uk-only.zarr
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2014-3000m-uk-only.zarr
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2015-3000m-uk-only.zarr
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2016-3000m-uk-only.zarr
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2017-3000m-uk-only.zarr
  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2018-3000m-uk-only.zarr
  history_mins: 165
  forecast_mins: 180
  sample_freq_mins: 15
  train_periods:
  - - 2008-01-01 00:00
    - 2017-12-31 23:55
  val_periods:
  - - 2018-01-01 00:00
    - 2018-12-31 23:55
  num_workers: 8
  prefetch_factor: 2
  batch_size: 2
  pin_memory: true
  persistent_workers: false
  dataset_pickle_dir: /tmp/sat_pred_dataset_cache
  seed: 27267831
  channels:
    VIS006:
      mean: 8.0
      std: 11.4
      clip_min: 0
      clip_max: 100
      missing_value: -11.4
    VIS008:
      mean: 9.3
      std: 13.0
      clip_min: 0
      clip_max: 100
      missing_value: -13.0
    IR_016:
      mean: 7.4
      std: 10.5
      clip_min: 0
      clip_max: 100
      missing_value: -10.5
    IR_039:
      mean: 267.5
      std: 14.9
      clip_min: 200
      clip_max: 330
      missing_value: 185.1
    WV_062:
      mean: 229.5
      std: 4.8
      clip_min: 190
      clip_max: 270
      missing_value: 185.2
    WV_073:
      mean: 243.9
      std: 8.1
      clip_min: 190
      clip_max: 270
      missing_value: 181.9
    IR_087:
      mean: 261.3
      std: 15.8
      clip_min: 190
      clip_max: 310
      missing_value: 174.2
    IR_097:
      mean: 239.4
      std: 8.8
      clip_min: 190
      clip_max: 280
      missing_value: 181.2
    IR_108:
      mean: 263.5
      std: 16.0
      clip_min: 190
      clip_max: 320
      missing_value: 174.0
    IR_120:
      mean: 262.4
      std: 16.1
      clip_min: 190
      clip_max: 310
      missing_value: 173.9
    IR_134:
      mean: 246.1
      std: 10.2
      clip_min: 190
      clip_max: 280
      missing_value: 179.8
callbacks:
  early_stopping:
    _target_: lightning.pytorch.callbacks.EarlyStopping
    monitor: MAE/val
    mode: min
    patience: 200
    min_delta: 0
  learning_rate_monitor:
    _target_: lightning.pytorch.callbacks.LearningRateMonitor
    logging_interval: epoch
  gradient_accumulation:
    _target_: lightning.pytorch.callbacks.GradientAccumulationScheduler
    _convert_: all
    scheduling:
      0: 16
      2: 16
      5: 16
      10: 16
  model_summary:
    _target_: lightning.pytorch.callbacks.ModelSummary
    max_depth: 4
  progress_bar:
    _target_: lightning.pytorch.callbacks.TQDMProgressBar
    refresh_rate: 50
  model_checkpoint:
    _target_: lightning.pytorch.callbacks.ModelCheckpoint
    monitor: MAE/val
    mode: min
    save_top_k: 1
    save_last: true
    every_n_epochs: 1
    verbose: false
    filename: epoch={epoch}-step={step}
    dirpath: checkpoints/simvp_v3_wsd
    auto_insert_metric_name: false
    save_on_train_epoch_end: false
logger:
  wandb:
    _target_: lightning.pytorch.loggers.wandb.WandbLogger
    project: sat_pred
    name: simvp_v3_wsd
    save_dir: ./
    offline: false
    id: null
    log_model: false
    prefix: ''
    job_type: train
    group: ''
    tags: []