未验证 提交 0856c186 编写于 作者: littletomatodonkey's avatar littletomatodonkey 提交者: GitHub

fix inference cls (#1123)

上级 40b71fac
...@@ -244,7 +244,7 @@ python3 python/predict_cls.py \ ...@@ -244,7 +244,7 @@ python3 python/predict_cls.py \
-c configs/inference_cls.yaml \ -c configs/inference_cls.yaml \
-o Global.infer_imgs=../dataset/flowers102/jpg/image_00001.jpg \ -o Global.infer_imgs=../dataset/flowers102/jpg/image_00001.jpg \
-o Global.inference_model_dir=../inference/ \ -o Global.inference_model_dir=../inference/ \
-o PostProcess.class_id_map_file=None -o PostProcess.Topk.class_id_map_file=None
其中: 其中:
......
...@@ -128,7 +128,7 @@ python3 -m paddle.distributed.launch \ ...@@ -128,7 +128,7 @@ python3 -m paddle.distributed.launch \
PaddleClas包含了自研的SSLD知识蒸馏方案,具体的内容可以参考[知识蒸馏章节](../advanced_tutorials/distillation/distillation.md), 本小节将尝试使用知识蒸馏技术对MobileNetV3_large_x1_0模型进行训练,使用`2.1.2小节`训练得到的ResNet50_vd模型作为蒸馏所用的教师模型,首先将`2.1.2小节`训练得到的ResNet50_vd模型保存到指定目录,脚本如下。 PaddleClas包含了自研的SSLD知识蒸馏方案,具体的内容可以参考[知识蒸馏章节](../advanced_tutorials/distillation/distillation.md), 本小节将尝试使用知识蒸馏技术对MobileNetV3_large_x1_0模型进行训练,使用`2.1.2小节`训练得到的ResNet50_vd模型作为蒸馏所用的教师模型,首先将`2.1.2小节`训练得到的ResNet50_vd模型保存到指定目录,脚本如下。
```shell ```shell
mkdir pretrained mkdir pretrained
cp -r output_CIFAR/ResNet50_vd/best_model.pdparams ./pretrained/ cp -r output_CIFAR/ResNet50_vd/best_model.pdparams ./pretrained/
``` ```
...@@ -256,5 +256,5 @@ PreProcess: ...@@ -256,5 +256,5 @@ PreProcess:
python3 python/predict_cls.py \ python3 python/predict_cls.py \
-c configs/inference_cls.yaml \ -c configs/inference_cls.yaml \
-o Global.infer_imgs=../dataset/CIFAR100/test/0/0001.png \ -o Global.infer_imgs=../dataset/CIFAR100/test/0/0001.png \
-o PostProcess.class_id_map_file=None -o PostProcess.Topk.class_id_map_file=None
``` ```
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册