From 84a1e2448d057f1e221a4c5f2b7482f423dc2217 Mon Sep 17 00:00:00 2001 From: Lingxi Xie <198808xc@gmail.com> Date: Thu, 2 Mar 2023 20:40:34 +0800 Subject: [PATCH] Update README.md --- README.md | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index 9697bf2..adb35c5 100644 --- a/README.md +++ b/README.md @@ -18,10 +18,10 @@ The downloaded files shall be organized as the following hierarchy: │ │ ├── input_surface.npy │ │ ├── input_upper.npy │ ├── output_data -│ ├── model_jit_cpu_1.onnx -│ ├── model_jit_cpu_3.onnx -│ ├── model_jit_cpu_6.onnx -│ ├── model_jit_cpu_24.onnx +│ ├── pangu_weather_1.onnx +│ ├── pangu_weather_3.onnx +│ ├── pangu_weather_6.onnx +│ ├── pangu_weather_24.onnx │ ├── inference_cpu.py │ ├── inference_gpu.py │ ├── inference_iterative.py @@ -43,13 +43,13 @@ pip install -r requirement_gpu.txt Please download the four pre-trained models (~1.1GB each) from Google drive or Baidu netdisk: -The 1-hour model (model_jit_cpu_1.onnx): [Google drive](https://drive.google.com/file/d/1fg5jkiN_5dHzKb-5H9Aw4MOmfILmeY-S/view?usp=share_link)/[Baidu netdisk](https://pan.baidu.com/s/1M7SAigVsCSH8hpw6DE8TDQ?pwd=ie0h) +The 1-hour model (pangu_weather_1.onnx): [Google drive](https://drive.google.com/file/d/1fg5jkiN_5dHzKb-5H9Aw4MOmfILmeY-S/view?usp=share_link)/[Baidu netdisk](https://pan.baidu.com/s/1M7SAigVsCSH8hpw6DE8TDQ?pwd=ie0h) -The 3-hour model (model_jit_cpu_3.onnx): [Google drive](https://drive.google.com/file/d/1EdoLlAXqE9iZLt9Ej9i-JW9LTJ9Jtewt/view?usp=share_link)/[Baidu netdisk](https://pan.baidu.com/s/197fZsoiCqZYzKwM7tyRrfg?pwd=gmcl) +The 3-hour model (pangu_weather_3.onnx): [Google drive](https://drive.google.com/file/d/1EdoLlAXqE9iZLt9Ej9i-JW9LTJ9Jtewt/view?usp=share_link)/[Baidu netdisk](https://pan.baidu.com/s/197fZsoiCqZYzKwM7tyRrfg?pwd=gmcl) -The 6-hour model (model_jit_cpu_6.onnx): [Google drive](https://drive.google.com/file/d/1a4XTktkZa5GCtjQxDJb_fNaqTAUiEJu4/view?usp=share_link)/[Baidu netdisk](https://pan.baidu.com/s/1q7IB7tNjqIwoGC7KVMPn4w?pwd=vxq3) +The 6-hour model (pangu_weather_6.onnx): [Google drive](https://drive.google.com/file/d/1a4XTktkZa5GCtjQxDJb_fNaqTAUiEJu4/view?usp=share_link)/[Baidu netdisk](https://pan.baidu.com/s/1q7IB7tNjqIwoGC7KVMPn4w?pwd=vxq3) -The 24-hour model (model_jit_cpu_24.onnx): [Google drive](https://drive.google.com/file/d/1lweQlxcn9fG0zKNW8ne1Khr9ehRTI6HP/view?usp=share_link)/[Baidu netdisk](https://pan.baidu.com/s/179q2gkz2BrsOR6g3yfTVQg?pwd=eajy) +The 24-hour model (pangu_weather.onnx): [Google drive](https://drive.google.com/file/d/1lweQlxcn9fG0zKNW8ne1Khr9ehRTI6HP/view?usp=share_link)/[Baidu netdisk](https://pan.baidu.com/s/179q2gkz2BrsOR6g3yfTVQg?pwd=eajy) These models are stored using the ONNX format, and thus can be used via different languages such as Python, C++, C#, Java, etc.