-
Notifications
You must be signed in to change notification settings - Fork 766
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
推理时动态Reshape试无法推理output shape的问题 #532
Comments
你可以试一下,不用动态reshape,直接用TNNTest测试你想要输入的大小,确定下试reshape的问题,还是这个尺寸本身就不行 |
模型本身是卷积网络,应该是支持动态shape的,模型在https://github.com/Tencent/TNN/issues/480里有,TNNTest是指TNN/test/test.cc吗? |
我用自己的程序,不用reshape,直接用其他shape的图片是不行的,必须是模型固定的shape1 * 3 * 960 * 1280,但是这不就是reshape的功能吗? |
对,就是test.cc,直接用scripts目录下的build_linux.sh,会编译出一个TNNTest,在linux环境下运行,可以指定参数为不同的shape |
是网络的问题,感谢 |
使用动态Reshape推理报错,我看是源码里这一段报出的错误,但是dim0[i] == 1 || dim1[i] == 1 || dim1[i] == dim0[i]应该是满足true的,麻烦看下,谢谢
报错如下:
![image](https://user-images.githubusercontent.com/13417277/98621999-7e14b780-2343-11eb-958a-39494d1a3bc2.png)
推理代码如下:
The text was updated successfully, but these errors were encountered: