- 博客(1)
- 资源 (4)
- 收藏
- 关注
RSS订阅原创 pytorch转tensorRT踩的几个小坑
1、版本不匹配[E] [TRT] Layer:Where_51's output can not be used as shape tensor.[E] [TRT] Network validation failed.[E] Engine creation failed.[E] Engine set up failed.这实际是由于pytorch与TensorRT版本不匹配，我的TensorRT是7.0，pytorch应该是1.4，但我用了1.7因此需要用1.7重新读取权重文件，然后用老
2020-12-20 16:34:10
4874
3
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人