华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
第一次用300I PRO这个硬件,现在我在GPU上yolov3模型训练完成,并进行推理。如果将训练好的模型部署在这个板卡上,除了将模型转换成.om格式,那么我的推理文件是自己手写吗?因为之前的推理文件是在pytorch框架下的,调的都是torch中的一些库,不太熟悉华为的这个推理文件该怎么修改,有没有大佬指点一下迷津?
我要发帖子
第一次用300I PRO这个硬件,现在我在GPU上yolov3模型训练完成,并进行推理。如果将训练好的模型部署在这个板卡上,除了将模型转换成.om格式,那么我的推理文件是自己手写吗?因为之前的推理文件是在pytorch框架下的,调的都是torch中的一些库,不太熟悉华为的这个推理文件该怎么修改,有没有大佬指点一下迷津?