├── .gitignore ├── README.md ├── era5_data ├── config.py ├── ordered_easydict.py ├── score.py ├── utils.py ├── utils_data.py └── utils_dist.py ├── fig ├── VIS.png ├── tab1.png ├── tab2.png └── vis_onnx.png ├── finetune ├── finetune_fully.py └── lora_tune.py ├── inference ├── inference_singleOutput.py ├── test_lora.py └── test_main.py ├── keys_all.csv └── models ├── layers.py ├── onnx2torch.py ├── pangu_model.py └── pangu_sample.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /era5_data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/era5_data/config.py -------------------------------------------------------------------------------- /era5_data/ordered_easydict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/era5_data/ordered_easydict.py -------------------------------------------------------------------------------- /era5_data/score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/era5_data/score.py -------------------------------------------------------------------------------- /era5_data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/era5_data/utils.py -------------------------------------------------------------------------------- /era5_data/utils_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/era5_data/utils_data.py -------------------------------------------------------------------------------- /era5_data/utils_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/era5_data/utils_dist.py -------------------------------------------------------------------------------- /fig/VIS.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/fig/VIS.png -------------------------------------------------------------------------------- /fig/tab1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/fig/tab1.png -------------------------------------------------------------------------------- /fig/tab2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/fig/tab2.png -------------------------------------------------------------------------------- /fig/vis_onnx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/fig/vis_onnx.png -------------------------------------------------------------------------------- /finetune/finetune_fully.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/finetune/finetune_fully.py -------------------------------------------------------------------------------- /finetune/lora_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/finetune/lora_tune.py -------------------------------------------------------------------------------- /inference/inference_singleOutput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/inference/inference_singleOutput.py -------------------------------------------------------------------------------- /inference/test_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/inference/test_lora.py -------------------------------------------------------------------------------- /inference/test_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/inference/test_main.py -------------------------------------------------------------------------------- /keys_all.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/keys_all.csv -------------------------------------------------------------------------------- /models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/models/layers.py -------------------------------------------------------------------------------- /models/onnx2torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/models/onnx2torch.py -------------------------------------------------------------------------------- /models/pangu_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/models/pangu_model.py -------------------------------------------------------------------------------- /models/pangu_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoshan2/pangu-pytorch/HEAD/models/pangu_sample.py --------------------------------------------------------------------------------