File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -782,7 +782,7 @@ solver = ppsci.solver.Solver(
782782 INFER: # (42)
783783 pretrained_model_path: " https://paddle-org.bj.bcebos.com/paddlescience/models/viv/viv_pretrained.pdparams" # (43)
784784 export_path: ./inference/viv # (44)
785- pdmodel_path: ${INFER.export_path} .pdmodel # (45)
785+ pdmodel_path: ${INFER.export_path} .json # (45)
786786 pdiparams_path: ${INFER.export_path} .pdiparams # (46)
787787 input_keys: ${MODEL.input_keys} # (47)
788788 output_keys: [" eta" , " f" ] # (48)
@@ -843,7 +843,7 @@ solver = ppsci.solver.Solver(
843843 42. ` INFER:` - 开始定义推理(推断)相关的设置。
844844 43. ` pretrained_model_path: " https://..." ` - 指定预训练模型的路径,可以是实际路径或 url,用于推理。
845845 44. ` export_path: ./inference/viv` - 设置模型导出路径,即推理所需的模型文件保存位置。
846- 45. ` pdmodel_path: ${INFER.export_path} .pdmodel ` - 指定Paddle模型的结构文件路径。
846+ 45. ` pdmodel_path: ${INFER.export_path} .json ` - 指定Paddle模型的结构文件路径,后缀名可以是 ` .json ` 或 ` .pdmodel ` 。
847847 46. ` pdiparams_path: ${INFER.export_path} .pdiparams` - 指定Paddle模型的参数文件路径。
848848 47. ` input_keys: ${MODEL.input_keys} ` - 推理时使用的输入键与模型定义时相同。
849849 48. ` output_keys: [" eta" , " f" ]` - 设置推理时的输出键,包括" eta" 和" f" 。
Original file line number Diff line number Diff line change @@ -360,7 +360,7 @@ pip install "paddle2onnx>=2.0.0"
360360
361361 少数案例尚未支持导出、推理功能,因此对应文档中未给出导出、推理命令。
362362
363- 首先需参考 [1 .2 模型导出](#12 ) 章节,从 `*.pdparams` 文件导出 `*.pdmodel `, `*.pdiparams` 两个文件。
363+ 首先需参考 [1 .2 模型导出](#12 ) 章节,从 `*.pdparams` 文件导出 `*.json `, `*.pdiparams` 两个文件。
364364
365365以 [Aneurysm](./examples/aneurysm.md) 案例为例,假设导出后的模型文件以 `./inference/aneurysm.*` 的形式保存,则推理代码示例如下。
366366
@@ -478,7 +478,7 @@ PaddleScience 提供了多种推理配置组合,可通过命令行进行组合
478478
479479 ONNX 是微软开源的深度学习推理框架,PaddleScience 支持了 ONNX 推理功能。
480480
481- 首先按照 [1.2.2 ONNX 推理模型导出](# 122-onnx) 章节将 `*.pdmodel ` 和 `*.pdiparams` 转换为 `*.onnx` 文件,
481+ 首先按照 [1.2.2 ONNX 推理模型导出](# 122-onnx) 章节将 `*.json ` 和 `*.pdiparams` 转换为 `*.onnx` 文件,
482482 然后根据硬件环境,安装 CPU 或 GPU 版的 onnxruntime:
483483
484484 ` ` ` sh
You can’t perform that action at this time.
0 commit comments