├── LICENSE ├── README.md ├── aardvark ├── architectures.py ├── data_shapes.py ├── e2e_model.py ├── e2e_train.py ├── finetune.py ├── generate_initial_condition_single.py ├── loader.py ├── loader_utils_new.py ├── loss_functions.py ├── misc_downscaling_functionality.py ├── models.py ├── set_convs.py ├── train_module.py ├── trainer.py ├── unet_wrap_padding.py ├── utils.py └── vit.py ├── data ├── grid_lon_lat │ ├── era5_x_1.npy │ └── era5_y_1.npy ├── norm_factors │ ├── mean_4u_1.npy │ ├── mean_diff_4u_1.npy │ ├── mean_hadisd_tas.npy │ ├── mean_hadisd_ws.npy │ ├── std_4u_1.npy │ ├── std_diff_4u_1.npy │ ├── std_hadisd_tas.npy │ └── std_hadisd_ws.npy └── sample_data_final.pkl ├── environment.yml ├── notebooks ├── data_demo.ipynb ├── e2e_finetune_demo.ipynb └── forecast_demo.ipynb └── training ├── downscaling_run_pipeline.sh ├── finetune.sh ├── test └── config.pkl ├── train_decoder.sh ├── train_e2e.sh ├── train_encoder.sh └── train_processor.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/README.md -------------------------------------------------------------------------------- /aardvark/architectures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/architectures.py -------------------------------------------------------------------------------- /aardvark/data_shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/data_shapes.py -------------------------------------------------------------------------------- /aardvark/e2e_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/e2e_model.py -------------------------------------------------------------------------------- /aardvark/e2e_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/e2e_train.py -------------------------------------------------------------------------------- /aardvark/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/finetune.py -------------------------------------------------------------------------------- /aardvark/generate_initial_condition_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/generate_initial_condition_single.py -------------------------------------------------------------------------------- /aardvark/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/loader.py -------------------------------------------------------------------------------- /aardvark/loader_utils_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/loader_utils_new.py -------------------------------------------------------------------------------- /aardvark/loss_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/loss_functions.py -------------------------------------------------------------------------------- /aardvark/misc_downscaling_functionality.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/misc_downscaling_functionality.py -------------------------------------------------------------------------------- /aardvark/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/models.py -------------------------------------------------------------------------------- /aardvark/set_convs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/set_convs.py -------------------------------------------------------------------------------- /aardvark/train_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/train_module.py -------------------------------------------------------------------------------- /aardvark/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/trainer.py -------------------------------------------------------------------------------- /aardvark/unet_wrap_padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/unet_wrap_padding.py -------------------------------------------------------------------------------- /aardvark/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/utils.py -------------------------------------------------------------------------------- /aardvark/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/aardvark/vit.py -------------------------------------------------------------------------------- /data/grid_lon_lat/era5_x_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/grid_lon_lat/era5_x_1.npy -------------------------------------------------------------------------------- /data/grid_lon_lat/era5_y_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/grid_lon_lat/era5_y_1.npy -------------------------------------------------------------------------------- /data/norm_factors/mean_4u_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/norm_factors/mean_4u_1.npy -------------------------------------------------------------------------------- /data/norm_factors/mean_diff_4u_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/norm_factors/mean_diff_4u_1.npy -------------------------------------------------------------------------------- /data/norm_factors/mean_hadisd_tas.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/norm_factors/mean_hadisd_tas.npy -------------------------------------------------------------------------------- /data/norm_factors/mean_hadisd_ws.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/norm_factors/mean_hadisd_ws.npy -------------------------------------------------------------------------------- /data/norm_factors/std_4u_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/norm_factors/std_4u_1.npy -------------------------------------------------------------------------------- /data/norm_factors/std_diff_4u_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/norm_factors/std_diff_4u_1.npy -------------------------------------------------------------------------------- /data/norm_factors/std_hadisd_tas.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/norm_factors/std_hadisd_tas.npy -------------------------------------------------------------------------------- /data/norm_factors/std_hadisd_ws.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/norm_factors/std_hadisd_ws.npy -------------------------------------------------------------------------------- /data/sample_data_final.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/data/sample_data_final.pkl -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/environment.yml -------------------------------------------------------------------------------- /notebooks/data_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/notebooks/data_demo.ipynb -------------------------------------------------------------------------------- /notebooks/e2e_finetune_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/notebooks/e2e_finetune_demo.ipynb -------------------------------------------------------------------------------- /notebooks/forecast_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/notebooks/forecast_demo.ipynb -------------------------------------------------------------------------------- /training/downscaling_run_pipeline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/training/downscaling_run_pipeline.sh -------------------------------------------------------------------------------- /training/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/training/finetune.sh -------------------------------------------------------------------------------- /training/test/config.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/training/test/config.pkl -------------------------------------------------------------------------------- /training/train_decoder.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/training/train_decoder.sh -------------------------------------------------------------------------------- /training/train_e2e.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/training/train_e2e.sh -------------------------------------------------------------------------------- /training/train_encoder.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/training/train_encoder.sh -------------------------------------------------------------------------------- /training/train_processor.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/anna-allen/aardvark-weather-public/HEAD/training/train_processor.sh --------------------------------------------------------------------------------