【PPSCI Export&Infer No.34】 #1129
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
Thanks for your contribution! |
| prefix="wind", | ||
| ) | ||
| visualizer_weather.save(cfg.INFER.export_path, vis_dict) | ||
| save_path = save_path = f"{cfg.INFER.export_path}/predict.npz" |
There was a problem hiding this comment.
Suggested change
| save_path = save_path = f"{cfg.INFER.export_path}/predict.npz" | |
| save_path = osp.join(cfg.INFER.export_path, "predict.npz") |
|
|
||
| # general settings | ||
| mode: train # running mode: train/eval | ||
| mode: infer # running mode: train/eval |
There was a problem hiding this comment.
Suggested change
| mode: infer # running mode: train/eval | |
| mode: train # running mode: train/eval/export/infer |
| # distributed under the License is distributed on an "AS IS" BASIS, | ||
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| # See the License for the specific language governing permissions and | ||
| # limitations under the License. |
There was a problem hiding this comment.
Suggested change
| # limitations under the License. | |
| # limitations under the License. | |
好的 |
Comment on lines
29
to
49
| === "模型导出命令" | ||
|
|
||
| ``` sh | ||
| # 风速预训练模型导出 | ||
| python train_pretrain.py mode==export | ||
| # 风速微调模型导出 | ||
| python train_finetune.py mode==export | ||
| # 降水量模型导出 | ||
| python train_precip.py mode==export | ||
| ``` | ||
|
|
||
| === "模型推理命令" | ||
|
|
||
| ``` sh | ||
| # 风速预训练模型推理 | ||
| python train_pretrain.py mode==infer | ||
| # 风速微调模型推理 | ||
| python train_finetune.py mode==infer | ||
| # 降水量模型推理 | ||
| python train_precip.py mode==infer | ||
| ``` |
docs/zh/examples/fourcastnet.md
Outdated
Comment on lines
43
to
48
| # 风速预训练模型推理 | ||
| python train_pretrain.py mode==infer | ||
| # 风速微调模型推理 | ||
| python train_finetune.py mode==infer | ||
| # 降水量模型推理 | ||
| python train_precip.py mode==infer |
There was a problem hiding this comment.
Suggested change
| # 风速预训练模型推理 | |
| python train_pretrain.py mode==infer | |
| # 风速微调模型推理 | |
| python train_finetune.py mode==infer | |
| # 降水量模型推理 | |
| python train_precip.py mode==infer | |
| # 下载风速预测小样本数据 | |
| wget -nc https://paddle-org.bj.bcebos.com/paddlescience%2Fdatasets%2FFourcastNet%2F2018-04-04_n6_precip.npy -P ./datasets/era5/test/ | |
| wget -nc https://paddle-org.bj.bcebos.com/paddlescience%2Fdatasets%2FFourcastNet%2F2018-04-04_n6.npy -P ./datasets/era5/test/ | |
| # 风速预训练模型推理 | |
| python train_pretrain.py mode=infer | |
| # 风速微调模型推理 | |
| python train_finetune.py mode=infer | |
| # 降水量模型推理 | |
| # 下载降水量预测小样本数据 | |
| wget -nc https://paddle-org.bj.bcebos.com/paddlescience%2Fdatasets%2FFourcastNet%2F2018-09-08_n32.npy -P ./datasets/era5/test/ | |
| python train_precip.py mode=infer |
examples/fourcastnet/train_precip.py
Outdated
Comment on lines
424
to
426
| xticks=np.linspace(0, 1439, 13), | ||
| xticklabels=[str(i) for i in range(360, -1, -30)], | ||
| yticks=np.linspace(0, 719, 7), |
There was a problem hiding this comment.
1439, 719这类magic number建议使用cfg.IMG_W-1之类的变量代替
examples/fourcastnet/train_precip.py
Outdated
Comment on lines
366
to
367
|
|
||
| output_keys = tuple(f"output_{i}" for i in range(cfg.INFER.num_timestamps)) |
There was a problem hiding this comment.
Suggested change
| output_keys = tuple(f"output_{i}" for i in range(cfg.INFER.num_timestamps)) | |
| output_keys = tuple(f"output_{i}" for i in range(cfg.INFER.num_timestamps)) |
|
hi, @mayiru1021
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
为fourcastnet案例的pretrain.py文件添加export与inference功能
PR changes
添加 export函数，inference函数
Describe
添加 export函数，inference函数，在yaml文件中添加了对应设置，实现了可视化功能。