但我看yolov5的例子可以在mlu上加载权重啊,但我yolov7照着写就是报错,这可能是什么原因导致的呢
您好,使用原生pytorch模型进行magicmind模型迁移,是需要在cpu上加载模型权重的;建议添加map_location=‘cpu’另外,yolov7的magicmind移植,可以参考这个demo:https://gitee.com/cambricon/magicmind_cloud/tree/master/buildin/cv/detection/yolov7_pytorch展开
但我看yolov5的例子可以在mlu上加载权重啊,但我yolov7照着写就是报错,这可能是什么原因导致的呢
https://gitee.com/cambricon/practices/tree/master/projects/pytorch_yolov5_inference展开
您好,使用原生pytorch模型进行magicmind模型迁移,是需要在cpu上加载模型权重的;建议添加map_location=‘cpu’
另外,yolov7的magicmind移植,可以参考这个demo:
https://gitee.com/cambricon/magicmind_cloud/tree/master/buildin/cv/detection/yolov7_pytorch
请登录后评论