TNN: developed by Tencent Youtu Lab and Guangying Lab, a uniform deep learning inference framework for mobile、desktop and server. TNN is distinguished by several outstanding features, including its cross-platform capability, high performance, model compression and code pruning. Based on ncnn and Rapidnet, TNN further strengthens the support and performance optimization for mobile devices, and also draws on the advantages of good extensibility and high performance from existed open source efforts. TNN has been deployed in multiple Apps from Tencent, such as Mobile QQ, Weishi, Pitu, etc. Contributions are welcome to work in collaborative with us and make TNN a better framework.
Other
4.37k
stars
765
forks
source link
SqueezeLayer has invalid input axes when convert lstm onnx to tnn model #1960
1. 使用环境（environment）
OS Version:
2. Github版本
commit(optional):
3. 详细描述bug 情况 (Describe the bug) A clear and concise description of what the bug is.
3层LSTM结构，将隐藏层拿出来保存用作序列模型。转换TNN的时候报错：ConstFolder Init Error: code: 0x1000 msg: SqueezeLayer has invalid input axes
模型文件见： 链接：https://pan.baidu.com/s/1ATR6mtytKvv-60ZN2IJ5-g 提取码：2fwm
4. 日志（Log）
日志见： 链接：https://pan.baidu.com/s/1ATR6mtytKvv-60ZN2IJ5-g 提取码：2fwm
5. 截图（Screenshots） If applicable, add screenshots to help explain your problem.