From a1f185a6ebc14b7a2598507e0e80be3c8597f872 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Wed, 22 Apr 2020 05:55:28 +0000 Subject: [PATCH] fix shell in getting_started --- docs/zh_CN/tutorials/getting_started.md | 6 +++--- tools/export_model.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/zh_CN/tutorials/getting_started.md b/docs/zh_CN/tutorials/getting_started.md index 0a3884ce..aee6351e 100644 --- a/docs/zh_CN/tutorials/getting_started.md +++ b/docs/zh_CN/tutorials/getting_started.md @@ -75,9 +75,9 @@ PaddlePaddle提供三种方式进行预测推理，接下来介绍如何用预 ```bash python tools/export_model.py \ - -model=模型名字 \ - -pretrained_model=预训练模型路径 \ - -output_path=预测模型保存路径 + --model=模型名字 \ + --pretrained_model=预训练模型路径 \ + --output_path=预测模型保存路径 ``` 之后，通过预测引擎进行推理： diff --git a/tools/export_model.py b/tools/export_model.py index d99f14a9..2cc32b3e 100644 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -24,7 +24,7 @@ def parse_args(): parser.add_argument("-m", "--model", type=str) parser.add_argument("-p", "--pretrained_model", type=str) parser.add_argument("-o", "--output_path", type=str) - parser.add_argument("--class_dim", type=int) + parser.add_argument("--class_dim", type=int, default=1000) return parser.parse_args() -- GitLab