| cluster: |
| wandb_mode: offline |
| use_custom_requeue: false |
| precision: bf16-mixed |
| batch_size: 1 |
| gpus: 4 |
| num_workers: 8 |
| blosc_nthreads: 3 |
| num_nodes: 1 |
| dataloader: |
| dataset: |
| _target_: archesweathersr.dataloaders.era5_hdf5.ERA5Downscaling |
| lowres_path: era5/240x121/weatherbench2/yearly |
| highres_path: era5/1440x721/weatherbench2/yearly |
| lead_time_hours: 24 |
| norm_scheme: era5 |
| domain: train |
| load_prev: false |
| multistep: 0 |
| validation_args: |
| domain: val_z0012 |
| test_args: |
| domain: test_z0012 |
| module: |
| metrics: |
| era5_ensemble_metrics: |
| _target_: geoarches.metrics.ensemble_metrics.Era5EnsembleMetrics |
| lead_time_hours: 24 |
| project: super-resolution |
| module: |
| _target_: archesweathersr.lightning_modules.sr_flow_matching.DownscalingDiffusionModule |
| name: archesweathersr-1 |
| cond_dim: 256 |
| scheduler: euler |
| prediction_type: sample |
| conditional: '' |
| interp_args: |
| mode: bicubic |
| align_corners: false |
| state_normalization: residual |
| pow: 2.0 |
| lr: 0.0003 |
| betas: |
| - 0.9 |
| - 0.98 |
| weight_decay: 0.01 |
| num_warmup_steps: 5000 |
| num_training_steps: 75000 |
| val: |
| sample_every_n_epochs: 5 |
| num_members: 2 |
| metrics: |
| - _target_: geoarches.metrics.ensemble_metrics.Era5EnsembleMetrics |
| lead_time_hours: 24 |
| metrics_kwargs: |
| save_memory: true |
| inference: |
| num_steps: 25 |
| num_members: 2 |
| cf_guidance: 1 |
| s_churn: 0.0 |
| save_test_outputs: false |
| metrics: |
| era5_ensemble_metrics: |
| _target_: geoarches.metrics.ensemble_metrics.Era5EnsembleMetrics |
| lead_time_hours: 24 |
| metrics_kwargs: |
| save_memory: true |
| backbone: |
| _target_: archesweathersr.backbones.archesweather.ArchesWeatherCondBackbone |
| tensor_size: |
| - 8 |
| - 180 |
| - 360 |
| emb_dim: 192 |
| cond_dim: 256 |
| window_size: |
| - 1 |
| - 6 |
| - 12 |
| droppath_coeff: 0.2 |
| dropout: 0 |
| depth_multiplier: 1 |
| use_skip: true |
| first_interaction_layer: linear |
| axis_attn: true |
| mlp_layer: swiglu |
| mlp_ratio: 4.0 |
| gradient_checkpointing: true |
| embedder: |
| _target_: archesweathersr.backbones.archesweather.WeatherEncodeDecodeLayer |
| img_size: |
| - 13 |
| - 721 |
| - 1440 |
| emb_dim: 192 |
| out_emb_dim: 384 |
| patch_size: |
| - 2 |
| - 4 |
| - 4 |
| surface_ch: 4 |
| level_ch: 6 |
| n_concatenated_states: 1 |
| log: true |
| entity: null |
| project: super-resolution |
| name: archesweathersr-1 |
| exp_dir: runs/archesweathersr-1 |
| resume: true |
| seed: None |
| max_steps: 75000 |
| batch_size: 1 |
| save_step_frequency: 2500 |
| log_freq: 100 |
| limit_val_batches: 1.0 |
| accumulate_grad_batches: 1 |
| mode: train |
|
|
