openclimatefix-models/cloudcasting_uk

1model_name: simvp_v3_wsd2seed: 272678313trainer:4  _target_: lightning.pytorch.trainer.trainer.Trainer5  accelerator: gpu6  devices:7  - 18  precision: 16-mixed9  gradient_clip_val: 0.510  detect_anomaly: false11  overfit_batches: 0.012  min_epochs: null13  max_epochs: 8014  reload_dataloaders_every_n_epochs: 015  num_sanity_val_steps: 816  fast_dev_run: false17  accumulate_grad_batches: 118  limit_train_batches: 1200019  limit_val_batches: 40020  log_every_n_steps: 5021model:22  _target_: sat_pred.training_module.TrainingModule23  model:24    _target_: sat_pred.models.simvp_model.SimVP25    num_channels: 1126    history_len: 1227    forecast_len: 1228    hid_S: 6429    hid_T: 51230    N_S: 431    N_T: 432  restore_optimizer_state: true33  optimizer:34    _target_: sat_pred.optimizers.AdamWWSD35    lr: 0.000536    warmup_epochs: 037    decay_epochs: 3038    total_epochs: 8039    final_factor: 0.0240    weight_decay: 0.0141  target_loss: MAE42  video_plot_t0_times:43  - '2018-06-08T12:00:00'44  - '2018-12-26T12:00:00'45  - '2018-06-14T12:00:00'46  - '2018-05-08T12:00:00'47  - '2018-05-12T12:00:00'48  - '2018-01-29T12:00:00'49  video_crop_plots: null50datamodule:51  _target_: sat_pred.dataset.SatelliteDataModule52  zarr_path:53  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2008-3000m-uk-only.zarr54  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2009-3000m-uk-only.zarr55  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2010-3000m-uk-only.zarr56  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2011-3000m-uk-only.zarr57  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2012-3000m-uk-only.zarr58  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2013-3000m-uk-only.zarr59  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2014-3000m-uk-only.zarr60  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2015-3000m-uk-only.zarr61  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2016-3000m-uk-only.zarr62  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2017-3000m-uk-only.zarr63  - /mnt/storage_u2_30tb_a/ml_training_zarrs/sat/uk_sat_crops/v6_symlink_minimal/rss-2018-3000m-uk-only.zarr64  history_mins: 16565  forecast_mins: 18066  sample_freq_mins: 1567  train_periods:68  - - 2008-01-01 00:0069    - 2017-12-31 23:5570  val_periods:71  - - 2018-01-01 00:0072    - 2018-12-31 23:5573  num_workers: 874  prefetch_factor: 275  batch_size: 276  pin_memory: true77  persistent_workers: false78  dataset_pickle_dir: /tmp/sat_pred_dataset_cache79  seed: 2726783180  channels:81    VIS006:82      mean: 8.083      std: 11.484      clip_min: 085      clip_max: 10086      missing_value: -11.487    VIS008:88      mean: 9.389      std: 13.090      clip_min: 091      clip_max: 10092      missing_value: -13.093    IR_016:94      mean: 7.495      std: 10.596      clip_min: 097      clip_max: 10098      missing_value: -10.599    IR_039:100      mean: 267.5101      std: 14.9102      clip_min: 200103      clip_max: 330104      missing_value: 185.1105    WV_062:106      mean: 229.5107      std: 4.8108      clip_min: 190109      clip_max: 270110      missing_value: 185.2111    WV_073:112      mean: 243.9113      std: 8.1114      clip_min: 190115      clip_max: 270116      missing_value: 181.9117    IR_087:118      mean: 261.3119      std: 15.8120      clip_min: 190121      clip_max: 310122      missing_value: 174.2123    IR_097:124      mean: 239.4125      std: 8.8126      clip_min: 190127      clip_max: 280128      missing_value: 181.2129    IR_108:130      mean: 263.5131      std: 16.0132      clip_min: 190133      clip_max: 320134      missing_value: 174.0135    IR_120:136      mean: 262.4137      std: 16.1138      clip_min: 190139      clip_max: 310140      missing_value: 173.9141    IR_134:142      mean: 246.1143      std: 10.2144      clip_min: 190145      clip_max: 280146      missing_value: 179.8147callbacks:148  early_stopping:149    _target_: lightning.pytorch.callbacks.EarlyStopping150    monitor: MAE/val151    mode: min152    patience: 200153    min_delta: 0154  learning_rate_monitor:155    _target_: lightning.pytorch.callbacks.LearningRateMonitor156    logging_interval: epoch157  gradient_accumulation:158    _target_: lightning.pytorch.callbacks.GradientAccumulationScheduler159    _convert_: all160    scheduling:161      0: 16162      2: 16163      5: 16164      10: 16165  model_summary:166    _target_: lightning.pytorch.callbacks.ModelSummary167    max_depth: 4168  progress_bar:169    _target_: lightning.pytorch.callbacks.TQDMProgressBar170    refresh_rate: 50171  model_checkpoint:172    _target_: lightning.pytorch.callbacks.ModelCheckpoint173    monitor: MAE/val174    mode: min175    save_top_k: 1176    save_last: true177    every_n_epochs: 1178    verbose: false179    filename: epoch={epoch}-step={step}180    dirpath: checkpoints/simvp_v3_wsd181    auto_insert_metric_name: false182    save_on_train_epoch_end: false183logger:184  wandb:185    _target_: lightning.pytorch.loggers.wandb.WandbLogger186    project: sat_pred187    name: simvp_v3_wsd188    save_dir: ./189    offline: false190    id: null191    log_model: false192    prefix: ''193    job_type: train194    group: ''195    tags: []196