产品型号MLU270CNML: 7.10.2 ba2048758CNRT: 4.10.1 a884a9apython3+torch1.3.0a0+b8d5360展开
产品型号MLU270CNML: 7.10.2 ba2048758CNRT: 4.10.1 a884a9apython3+torch1.3.0a0+b8d5360展开
您好,测试样例验证步骤:
1、声明TORCH_HOME(存放模型路径)和VOC_DEVKIT(数据集路径)环境变量;
2、下载模型到$TORCH_HOME/origin/checkpoints/路径,下载链接:wget https://s3.amazonaws.com/amdegroot-models/ssd300_mAP_77.43_v2.pth
3、模型量化(可参考ssd路径下的README.md)
python /torch/examples/online/ssd/eval.py --voc_root $VOC_DEVKIT--quantization true --image_number 50 --quantized_mode 1 --quantized_model_path $TORCH_HOME/int8/checkpoints
该步骤会在$TORCH_HOME/int8/checkpoints/路径下生成量化后模型。
4、在线推理
bash /torch/examples/online/ssd/run_test_ssd.sh
5、离线推理并生成离线模型
bash /torch/examples/offline/ssd/run_all_offline_mc.sh 1 1
请登录后评论