pytorch转tensorrt遇到的一些问题？
1.AttributeError:module 'tensorrt' has no attribute 'PaddingMode'解决办法：是tensorrt版本问题，可以将版本升级到5.1+就可以解决这个问题。2.ValueError: negative shift count解决办法：这个是由于softmax参数设置造成的错误，之前softmax（dim=-1），将dim = ...
·
1. AttributeError:module 'tensorrt' has no attribute 'PaddingMode'
解决办法：是tensorrt版本问题，可以将版本升级到5.1+就可以解决这个问题。
2. ValueError: negative shift count
解决办法：这个是由于softmax参数设置造成的错误，之前softmax（dim=-1），将dim = 1就可以解决这个问题。
汇聚全球AI编程工具，助力开发者即刻编程。
更多推荐
2
0- 0
已为社区贡献3条内容
所有评论(0)