打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
你好,我司在tf 模型.pb 从x86+ gpu环境 移植适配到寒武纪 mlu270上遇到几个棘手的问题,问题总结如下:
1) 270上 ,.pb移植过程生成的.cambricon 模型维度参数不对, NCWH分别是1,32681, 64, 3096296608, 这个结果是错误的。
同一个pb模型在100上,生成的.cambricon模型维度参数NCWH是(1,1,9,1),这个结果是对的。
(模式输入维度:1*9)
请问遇到这种情况,该如何解决? .pb文件请下载附近(test.rar)
2)270上,利用上面生成的.cambricon模型部署,运行test_forward_offline.cpp ,用测试用例(详见附近.c),测试得到的输出结果错误。
同一个pb模型,在100上离线生成.cambricon,运行test_forward_offline.cpp,同一组测试输入用例,测试输出结果是正确的。
我的微信号:sunshine-jzj,请百忙中解决,谢谢!
热门帖子
精华帖子