×

签到

分享到微信

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

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

MLU270 tensorflow pb模型离线部署移植过程生成NCWH参数和测试结果异常 已解决 Sunshine2021-09-24 10:52:06 回复 3 查看 技术答疑
MLU270 tensorflow pb模型离线部署移植过程生成NCWH参数和测试结果异常
分享到:

你好,我司在tf 模型.pb 从x86+ gpu环境 移植适配到寒武纪 mlu270上遇到几个棘手的问题,问题总结如下:

1) 270上 ,.pb移植过程生成的.cambricon 模型维度参数不对,  NCWH分别是1,32681, 64,  3096296608, 这个结果是错误的。

     同一个pb模型在100上,生成的.cambricon模型维度参数NCWH是(1,1,9,1),这个结果是对的。

(模式输入维度:1*9)

image.png

请问遇到这种情况,该如何解决? .pb文件请下载附近(test.rar


2)270上,利用上面生成的.cambricon模型部署,运行test_forward_offline.cpp ,用测试用例(详见附近.c),测试得到的输出结果错误。 

同一个pb模型,在100上离线生成.cambricon,运行test_forward_offline.cpp,同一组测试输入用例,测试输出结果是正确的。

 

我的微信号:sunshine-jzj,请百忙中解决,谢谢!

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