fix test_tipc

pull/5361/head
LDOUBLEV 2022-01-26 11:43:56 +08:00
parent c83f579918
commit ee955e5bf7
1 changed files with 5 additions and 5 deletions

View File

@ -3,7 +3,7 @@
Linux端基础训练预测功能测试的主程序为test_train_python.sh可以测试基于Python的模型训练、评估等基本功能包括裁剪、量化、蒸馏训练。
![](./tipc_train.png)
![](./test_tipc/tipc_train.png)
测试链条如上图所示主要测试内容有带共享权重自定义OP的模型的正常训练和slim相关功能训练流程是否正常。
@ -28,23 +28,23 @@ pip3 install -r requirements.txt
- 模式1lite_train_lite_infer使用少量数据训练用于快速验证训练到预测的走通流程不验证精度和速度
```
bash test_tipc/test_train_python.sh ./test_tipc/ch_ppocr_mobile_v2.0_det/train_infer_python.txt 'lite_train_lite_infer'
bash test_tipc/test_train_python.sh ./test_tipc/train_infer_python.txt 'lite_train_lite_infer'
```
- 模式2whole_train_whole_infer使用全量数据训练用于快速验证训练到预测的走通流程验证模型最终训练精度
```
bash test_tipc/test_train_python.sh ./test_tipc/ch_ppocr_mobile_v2.0_det/train_infer_python.txt 'whole_train_whole_infer'
bash test_tipc/test_train_python.sh ./test_tipc/train_infer_python.txt 'whole_train_whole_infer'
```
如果是运行量化裁剪等训练方式,需要使用不同的配置文件。量化训练的测试指令如下:
```
bash test_tipc/test_train_python.sh ./test_tipc/ch_ppocr_mobile_v2.0_det/train_infer_python_PACT.txt 'lite_train_lite_infer'
bash test_tipc/test_train_python.sh ./test_tipc/train_infer_python_PACT.txt 'lite_train_lite_infer'
```
同理FPGM裁剪的运行方式如下
```
bash test_tipc/test_train_python.sh ./test_tipc/ch_ppocr_mobile_v2.0_det/train_infer_python_FPGM.txt 'lite_train_lite_infer'
bash test_tipc/test_train_python.sh ./test_tipc/train_infer_python_FPGM.txt 'lite_train_lite_infer'
```
运行相应指令后,在`test_tipc/output`文件夹下自动会保存运行日志。如'lite_train_lite_infer'模式运行后在test_tipc/extra_output文件夹有以下文件