├── .gitignore ├── .gitmodules ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── WD ├── WeatherBenchDataset.py ├── dataset_old.py ├── datasets.py ├── io.py ├── plotting.py ├── regridding.py └── utils.py ├── config ├── data.yaml ├── ds_format │ └── zarr.yaml ├── experiment │ ├── diffusion.yaml │ ├── diffusion_2csteps.yaml │ ├── diffusion_CosineAnnealing.yaml │ ├── diffusion_MSE_Loss.yaml │ ├── diffusion_MSE_Loss_more_patient_deeper.yaml │ ├── diffusion_deeper.yaml │ ├── diffusion_fourcast.yaml │ ├── diffusion_more_patient_deeper.yaml │ ├── diffusion_t2m_3day_highres.yaml │ ├── diffusion_t2m_5day_highres.yaml │ ├── diffusion_t_850_1day.yaml │ ├── diffusion_t_850_2day.yaml │ ├── diffusion_t_850_3day.yaml │ ├── diffusion_t_850_3day_highres.yaml │ ├── diffusion_t_850_4day.yaml │ ├── diffusion_t_850_5day.yaml │ ├── diffusion_t_850_5day_highres.yaml │ ├── diffusion_wider.yaml │ ├── diffusion_z_500_1day.yaml │ ├── diffusion_z_500_2day.yaml │ ├── diffusion_z_500_3day.yaml │ ├── diffusion_z_500_3day_highres.yaml │ ├── diffusion_z_500_4day.yaml │ ├── diffusion_z_500_5day.yaml │ ├── diffusion_z_500_5day_highres.yaml │ ├── fourcastnet.yaml │ ├── iterative_diffusion_reduced_set.yaml │ ├── iterative_diffusion_t_850.yaml │ ├── iterative_diffusion_t_850_highres.yaml │ ├── iterative_diffusion_z_500.yaml │ ├── iterative_diffusion_z_500_highres.yaml │ ├── iterative_diffusion_z_500_t_850.yaml │ ├── unet.yaml │ ├── unet_highres.yaml │ ├── unet_highres_t2m.yaml │ └── unet_highres_t2m_3day.yaml ├── inference.yaml ├── paths │ └── default_paths.yaml ├── template │ ├── iterative_rasp_thuerey.yaml │ ├── iterative_reduced_set.yaml │ ├── iterative_t_850.yaml │ ├── iterative_t_850_highres.yaml │ ├── iterative_z_500.yaml │ ├── iterative_z_500_highres.yaml │ ├── iterative_z_500_t_850.yaml │ ├── rasp_thuerey_highres_t2m_3day.yaml │ ├── rasp_thuerey_highres_t2m_5day.yaml │ ├── rasp_thuerey_highres_t_850_3day.yaml │ ├── rasp_thuerey_highres_t_850_5day.yaml │ ├── rasp_thuerey_highres_z_500_3day.yaml │ ├── rasp_thuerey_highres_z_500_5day.yaml │ ├── rasp_thuerey_t_850_1day.yaml │ ├── rasp_thuerey_t_850_2day.yaml │ ├── rasp_thuerey_t_850_3day.yaml │ ├── rasp_thuerey_t_850_4day.yaml │ ├── rasp_thuerey_t_850_5day.yaml │ ├── rasp_thuerey_z_500_1day.yaml │ ├── rasp_thuerey_z_500_2day.yaml │ ├── rasp_thuerey_z_500_3day.yaml │ ├── rasp_thuerey_z_500_3day_2csteps.yaml │ ├── rasp_thuerey_z_500_4day.yaml │ └── rasp_thuerey_z_500_5day.yaml └── train.yaml ├── env_data.yml ├── env_eval.yml ├── env_model.yml ├── images ├── chronologic_timesteps.jpg ├── ensemble_condition.jpg ├── ensemble_predictions.jpg ├── ensemble_stats.jpg ├── ensemble_std.jpg ├── heatwave_predictions_step_0.png ├── heatwave_predictions_step_1.png ├── heatwave_predictions_step_10.png ├── heatwave_predictions_step_2.png ├── heatwave_predictions_step_20.png ├── heatwave_predictions_step_32.png ├── heatwave_predictions_step_5.png ├── heatwave_true_anomaly.png ├── performance_leadtime_version_0.jpg ├── performance_leadtime_version_1.jpg ├── performance_leadtime_version_2.jpg ├── performance_leadtime_version_3.jpg ├── performance_leadtime_version_4.jpg ├── predictions.jpg ├── spectra.png ├── spectra_no_entries.png ├── t_850_lowres.gif ├── timeseries.jpg └── z_500_lowres.gif ├── nb_ensemble_eval.ipynb ├── nb_heatwave.ipynb ├── nb_performance_over_lead_time.ipynb ├── nb_results.ipynb ├── nb_spectral_analysis.ipynb ├── nb_test_predictions.ipynb ├── pyproject.toml ├── s10_write_predictions_vae.py ├── s11_train_LFD.py ├── s12_write_predictions_LFD.py ├── s13_write_predictions_iterative.py ├── s14_very_long_iterative_run.py ├── s1_write_dataset.py ├── s2_train_conditional_pixel_diffusion.py ├── s3_write_predictions_conditional_pixel_diffusion.py ├── s4_train_val_test.py ├── s5_train_FourCastNet.py ├── s6_write_predictions_FourCastNet.py ├── s7_train_unet.py ├── s8_write_predictions_unet.py ├── s9_train_vae.py ├── submit_script_10_inference_vae.sh ├── submit_script_11_train_LFD.sh ├── submit_script_12_inference_LFD.sh ├── submit_script_13_inference_iterative.sh ├── submit_script_14_inference_iterative_very_long.sh ├── submit_script_1_dataset_creation.sh ├── submit_script_2_run_model.sh ├── submit_script_3_inference.sh ├── submit_script_4_eval_epoch.sh ├── submit_script_5_train_FourCastNet.sh ├── submit_script_6_inference_fourcastnet.sh ├── submit_script_7_train_unet.sh ├── submit_script_8_inference_UNet.sh └── submit_script_9_train_vae.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/.gitmodules -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/README.md -------------------------------------------------------------------------------- /WD/WeatherBenchDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/WD/WeatherBenchDataset.py -------------------------------------------------------------------------------- /WD/dataset_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/WD/dataset_old.py -------------------------------------------------------------------------------- /WD/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/WD/datasets.py -------------------------------------------------------------------------------- /WD/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/WD/io.py -------------------------------------------------------------------------------- /WD/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/WD/plotting.py -------------------------------------------------------------------------------- /WD/regridding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/WD/regridding.py -------------------------------------------------------------------------------- /WD/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/WD/utils.py -------------------------------------------------------------------------------- /config/data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/data.yaml -------------------------------------------------------------------------------- /config/ds_format/zarr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/ds_format/zarr.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_2csteps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_2csteps.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_CosineAnnealing.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_CosineAnnealing.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_MSE_Loss.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_MSE_Loss.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_MSE_Loss_more_patient_deeper.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_MSE_Loss_more_patient_deeper.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_deeper.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_deeper.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_fourcast.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_fourcast.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_more_patient_deeper.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_more_patient_deeper.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_t2m_3day_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_t2m_3day_highres.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_t2m_5day_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_t2m_5day_highres.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_t_850_1day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_t_850_1day.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_t_850_2day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_t_850_2day.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_t_850_3day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_t_850_3day.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_t_850_3day_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_t_850_3day_highres.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_t_850_4day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_t_850_4day.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_t_850_5day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_t_850_5day.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_t_850_5day_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_t_850_5day_highres.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_wider.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_wider.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_z_500_1day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_z_500_1day.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_z_500_2day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_z_500_2day.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_z_500_3day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_z_500_3day.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_z_500_3day_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_z_500_3day_highres.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_z_500_4day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_z_500_4day.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_z_500_5day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_z_500_5day.yaml -------------------------------------------------------------------------------- /config/experiment/diffusion_z_500_5day_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/diffusion_z_500_5day_highres.yaml -------------------------------------------------------------------------------- /config/experiment/fourcastnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/fourcastnet.yaml -------------------------------------------------------------------------------- /config/experiment/iterative_diffusion_reduced_set.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/iterative_diffusion_reduced_set.yaml -------------------------------------------------------------------------------- /config/experiment/iterative_diffusion_t_850.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/iterative_diffusion_t_850.yaml -------------------------------------------------------------------------------- /config/experiment/iterative_diffusion_t_850_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/iterative_diffusion_t_850_highres.yaml -------------------------------------------------------------------------------- /config/experiment/iterative_diffusion_z_500.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/iterative_diffusion_z_500.yaml -------------------------------------------------------------------------------- /config/experiment/iterative_diffusion_z_500_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/iterative_diffusion_z_500_highres.yaml -------------------------------------------------------------------------------- /config/experiment/iterative_diffusion_z_500_t_850.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/iterative_diffusion_z_500_t_850.yaml -------------------------------------------------------------------------------- /config/experiment/unet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/unet.yaml -------------------------------------------------------------------------------- /config/experiment/unet_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/unet_highres.yaml -------------------------------------------------------------------------------- /config/experiment/unet_highres_t2m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/unet_highres_t2m.yaml -------------------------------------------------------------------------------- /config/experiment/unet_highres_t2m_3day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/experiment/unet_highres_t2m_3day.yaml -------------------------------------------------------------------------------- /config/inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/inference.yaml -------------------------------------------------------------------------------- /config/paths/default_paths.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/paths/default_paths.yaml -------------------------------------------------------------------------------- /config/template/iterative_rasp_thuerey.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/iterative_rasp_thuerey.yaml -------------------------------------------------------------------------------- /config/template/iterative_reduced_set.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/iterative_reduced_set.yaml -------------------------------------------------------------------------------- /config/template/iterative_t_850.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/iterative_t_850.yaml -------------------------------------------------------------------------------- /config/template/iterative_t_850_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/iterative_t_850_highres.yaml -------------------------------------------------------------------------------- /config/template/iterative_z_500.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/iterative_z_500.yaml -------------------------------------------------------------------------------- /config/template/iterative_z_500_highres.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/iterative_z_500_highres.yaml -------------------------------------------------------------------------------- /config/template/iterative_z_500_t_850.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/iterative_z_500_t_850.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_highres_t2m_3day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_highres_t2m_3day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_highres_t2m_5day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_highres_t2m_5day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_highres_t_850_3day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_highres_t_850_3day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_highres_t_850_5day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_highres_t_850_5day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_highres_z_500_3day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_highres_z_500_3day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_highres_z_500_5day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_highres_z_500_5day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_t_850_1day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_t_850_1day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_t_850_2day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_t_850_2day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_t_850_3day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_t_850_3day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_t_850_4day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_t_850_4day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_t_850_5day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_t_850_5day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_z_500_1day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_z_500_1day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_z_500_2day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_z_500_2day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_z_500_3day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_z_500_3day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_z_500_3day_2csteps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_z_500_3day_2csteps.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_z_500_4day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_z_500_4day.yaml -------------------------------------------------------------------------------- /config/template/rasp_thuerey_z_500_5day.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/template/rasp_thuerey_z_500_5day.yaml -------------------------------------------------------------------------------- /config/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/config/train.yaml -------------------------------------------------------------------------------- /env_data.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/env_data.yml -------------------------------------------------------------------------------- /env_eval.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/env_eval.yml -------------------------------------------------------------------------------- /env_model.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/env_model.yml -------------------------------------------------------------------------------- /images/chronologic_timesteps.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/chronologic_timesteps.jpg -------------------------------------------------------------------------------- /images/ensemble_condition.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/ensemble_condition.jpg -------------------------------------------------------------------------------- /images/ensemble_predictions.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/ensemble_predictions.jpg -------------------------------------------------------------------------------- /images/ensemble_stats.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/ensemble_stats.jpg -------------------------------------------------------------------------------- /images/ensemble_std.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/ensemble_std.jpg -------------------------------------------------------------------------------- /images/heatwave_predictions_step_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/heatwave_predictions_step_0.png -------------------------------------------------------------------------------- /images/heatwave_predictions_step_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/heatwave_predictions_step_1.png -------------------------------------------------------------------------------- /images/heatwave_predictions_step_10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/heatwave_predictions_step_10.png -------------------------------------------------------------------------------- /images/heatwave_predictions_step_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/heatwave_predictions_step_2.png -------------------------------------------------------------------------------- /images/heatwave_predictions_step_20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/heatwave_predictions_step_20.png -------------------------------------------------------------------------------- /images/heatwave_predictions_step_32.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/heatwave_predictions_step_32.png -------------------------------------------------------------------------------- /images/heatwave_predictions_step_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/heatwave_predictions_step_5.png -------------------------------------------------------------------------------- /images/heatwave_true_anomaly.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/heatwave_true_anomaly.png -------------------------------------------------------------------------------- /images/performance_leadtime_version_0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/performance_leadtime_version_0.jpg -------------------------------------------------------------------------------- /images/performance_leadtime_version_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/performance_leadtime_version_1.jpg -------------------------------------------------------------------------------- /images/performance_leadtime_version_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/performance_leadtime_version_2.jpg -------------------------------------------------------------------------------- /images/performance_leadtime_version_3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/performance_leadtime_version_3.jpg -------------------------------------------------------------------------------- /images/performance_leadtime_version_4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/performance_leadtime_version_4.jpg -------------------------------------------------------------------------------- /images/predictions.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/predictions.jpg -------------------------------------------------------------------------------- /images/spectra.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/spectra.png -------------------------------------------------------------------------------- /images/spectra_no_entries.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/spectra_no_entries.png -------------------------------------------------------------------------------- /images/t_850_lowres.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/t_850_lowres.gif -------------------------------------------------------------------------------- /images/timeseries.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/timeseries.jpg -------------------------------------------------------------------------------- /images/z_500_lowres.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/images/z_500_lowres.gif -------------------------------------------------------------------------------- /nb_ensemble_eval.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/nb_ensemble_eval.ipynb -------------------------------------------------------------------------------- /nb_heatwave.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/nb_heatwave.ipynb -------------------------------------------------------------------------------- /nb_performance_over_lead_time.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/nb_performance_over_lead_time.ipynb -------------------------------------------------------------------------------- /nb_results.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/nb_results.ipynb -------------------------------------------------------------------------------- /nb_spectral_analysis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/nb_spectral_analysis.ipynb -------------------------------------------------------------------------------- /nb_test_predictions.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/nb_test_predictions.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/pyproject.toml -------------------------------------------------------------------------------- /s10_write_predictions_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s10_write_predictions_vae.py -------------------------------------------------------------------------------- /s11_train_LFD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s11_train_LFD.py -------------------------------------------------------------------------------- /s12_write_predictions_LFD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s12_write_predictions_LFD.py -------------------------------------------------------------------------------- /s13_write_predictions_iterative.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s13_write_predictions_iterative.py -------------------------------------------------------------------------------- /s14_very_long_iterative_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s14_very_long_iterative_run.py -------------------------------------------------------------------------------- /s1_write_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s1_write_dataset.py -------------------------------------------------------------------------------- /s2_train_conditional_pixel_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s2_train_conditional_pixel_diffusion.py -------------------------------------------------------------------------------- /s3_write_predictions_conditional_pixel_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s3_write_predictions_conditional_pixel_diffusion.py -------------------------------------------------------------------------------- /s4_train_val_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s4_train_val_test.py -------------------------------------------------------------------------------- /s5_train_FourCastNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s5_train_FourCastNet.py -------------------------------------------------------------------------------- /s6_write_predictions_FourCastNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s6_write_predictions_FourCastNet.py -------------------------------------------------------------------------------- /s7_train_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s7_train_unet.py -------------------------------------------------------------------------------- /s8_write_predictions_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s8_write_predictions_unet.py -------------------------------------------------------------------------------- /s9_train_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/s9_train_vae.py -------------------------------------------------------------------------------- /submit_script_10_inference_vae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_10_inference_vae.sh -------------------------------------------------------------------------------- /submit_script_11_train_LFD.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_11_train_LFD.sh -------------------------------------------------------------------------------- /submit_script_12_inference_LFD.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_12_inference_LFD.sh -------------------------------------------------------------------------------- /submit_script_13_inference_iterative.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_13_inference_iterative.sh -------------------------------------------------------------------------------- /submit_script_14_inference_iterative_very_long.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_14_inference_iterative_very_long.sh -------------------------------------------------------------------------------- /submit_script_1_dataset_creation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_1_dataset_creation.sh -------------------------------------------------------------------------------- /submit_script_2_run_model.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_2_run_model.sh -------------------------------------------------------------------------------- /submit_script_3_inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_3_inference.sh -------------------------------------------------------------------------------- /submit_script_4_eval_epoch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_4_eval_epoch.sh -------------------------------------------------------------------------------- /submit_script_5_train_FourCastNet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_5_train_FourCastNet.sh -------------------------------------------------------------------------------- /submit_script_6_inference_fourcastnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_6_inference_fourcastnet.sh -------------------------------------------------------------------------------- /submit_script_7_train_unet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_7_train_unet.sh -------------------------------------------------------------------------------- /submit_script_8_inference_UNet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_8_inference_UNet.sh -------------------------------------------------------------------------------- /submit_script_9_train_vae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ECMWFCode4Earth/diffusion-models-for-weather-prediction/HEAD/submit_script_9_train_vae.sh --------------------------------------------------------------------------------