├── README.md ├── backbones ├── archesweather.py ├── archesweather_layers.py ├── dit.py └── weatherlearn_utils │ ├── README.md │ ├── crop.py │ ├── earth_position_index.py │ ├── pad.py │ ├── patch_embed.py │ ├── patch_recovery.py │ └── shift_window_mask.py ├── configs ├── cluster │ └── example-slurm.yaml ├── config.yaml ├── dataloader │ └── era5-w.yaml └── module │ └── forecast-archesweather.yaml ├── dataloaders ├── __init__.py ├── era5.py └── preprocessors.py ├── dl_era.py ├── evaluation └── deterministic_metrics.py ├── lightning_modules └── forecast.py ├── requirements.txt ├── scripts └── eval_multistep.py ├── stats ├── archesweather_constant_masks.pt ├── delta24_stats.pt ├── delta24_stats_with_w.pt ├── pangu_norm_stats2.pt └── pangu_norm_stats2_with_w.pt ├── submit.py └── train_hydra.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/README.md -------------------------------------------------------------------------------- /backbones/archesweather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/backbones/archesweather.py -------------------------------------------------------------------------------- /backbones/archesweather_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/backbones/archesweather_layers.py -------------------------------------------------------------------------------- /backbones/dit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/backbones/dit.py -------------------------------------------------------------------------------- /backbones/weatherlearn_utils/README.md: -------------------------------------------------------------------------------- 1 | this folder is copied from https://github.com/lizhuoq/WeatherLearn -------------------------------------------------------------------------------- /backbones/weatherlearn_utils/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/backbones/weatherlearn_utils/crop.py -------------------------------------------------------------------------------- /backbones/weatherlearn_utils/earth_position_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/backbones/weatherlearn_utils/earth_position_index.py -------------------------------------------------------------------------------- /backbones/weatherlearn_utils/pad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/backbones/weatherlearn_utils/pad.py -------------------------------------------------------------------------------- /backbones/weatherlearn_utils/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/backbones/weatherlearn_utils/patch_embed.py -------------------------------------------------------------------------------- /backbones/weatherlearn_utils/patch_recovery.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/backbones/weatherlearn_utils/patch_recovery.py -------------------------------------------------------------------------------- /backbones/weatherlearn_utils/shift_window_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/backbones/weatherlearn_utils/shift_window_mask.py -------------------------------------------------------------------------------- /configs/cluster/example-slurm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/configs/cluster/example-slurm.yaml -------------------------------------------------------------------------------- /configs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/configs/config.yaml -------------------------------------------------------------------------------- /configs/dataloader/era5-w.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/configs/dataloader/era5-w.yaml -------------------------------------------------------------------------------- /configs/module/forecast-archesweather.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/configs/module/forecast-archesweather.yaml -------------------------------------------------------------------------------- /dataloaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/dataloaders/__init__.py -------------------------------------------------------------------------------- /dataloaders/era5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/dataloaders/era5.py -------------------------------------------------------------------------------- /dataloaders/preprocessors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/dataloaders/preprocessors.py -------------------------------------------------------------------------------- /dl_era.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/dl_era.py -------------------------------------------------------------------------------- /evaluation/deterministic_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/evaluation/deterministic_metrics.py -------------------------------------------------------------------------------- /lightning_modules/forecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/lightning_modules/forecast.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/eval_multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/scripts/eval_multistep.py -------------------------------------------------------------------------------- /stats/archesweather_constant_masks.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/stats/archesweather_constant_masks.pt -------------------------------------------------------------------------------- /stats/delta24_stats.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/stats/delta24_stats.pt -------------------------------------------------------------------------------- /stats/delta24_stats_with_w.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/stats/delta24_stats_with_w.pt -------------------------------------------------------------------------------- /stats/pangu_norm_stats2.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/stats/pangu_norm_stats2.pt -------------------------------------------------------------------------------- /stats/pangu_norm_stats2_with_w.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/stats/pangu_norm_stats2_with_w.pt -------------------------------------------------------------------------------- /submit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/submit.py -------------------------------------------------------------------------------- /train_hydra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gcouairon/ArchesWeather/HEAD/train_hydra.py --------------------------------------------------------------------------------