pytorch轉tensorrt遇到的一些問題？

1. AttributeError:module 'tensorrt' has no attribute 'PaddingMode'

解決辦法：是tensorrt版本問題，可以將版本升級到5.1+就可以解決這個問題。

2. ValueError: negative shift count

解決辦法：這個是由於softmax參數設置造成的錯誤，之前softmax（dim=-1），將dim = 1就可以解決這個問題。