×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

yolov5 项目 训练的resnet18.pt 如何在寒武纪上量化移植 已完结 Cris1234562023-07-18 10:43:46 回复 2 查看 使用求助
yolov5 项目 训练的resnet18.pt 如何在寒武纪上量化移植
分享到:


yolov5 6.2之后,有个文件夹可训练分类模型,我训练出resnet18.pt 模型,按照文档中的要求改变存储格式。后续该如何量化和离线推理?
官方给的示例代码好像不是自己训练的resnet
自己训练的resnet 模型有相关的示例或者demo吗?还是可以自己通过pytorch_mlu 的API 自行编写脚本就能完成量化?

(某乎搬运的代码,我自己的docker 里没有class_Models 路径)

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭