├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── config.py ├── datamodule.py ├── distributed.py ├── environment.yml ├── finetune_gravity_wave.py ├── flux_prediction_prithvi_finetuning.gif ├── gravity_wave_model.py ├── inference.py └── scripts ├── train.pbs └── train.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/config.py -------------------------------------------------------------------------------- /datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/datamodule.py -------------------------------------------------------------------------------- /distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/distributed.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/environment.yml -------------------------------------------------------------------------------- /finetune_gravity_wave.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/finetune_gravity_wave.py -------------------------------------------------------------------------------- /flux_prediction_prithvi_finetuning.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/flux_prediction_prithvi_finetuning.gif -------------------------------------------------------------------------------- /gravity_wave_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/gravity_wave_model.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/inference.py -------------------------------------------------------------------------------- /scripts/train.pbs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/scripts/train.pbs -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/gravity-wave-finetuning/HEAD/scripts/train.sh --------------------------------------------------------------------------------