Closed niaoyu closed 3 years ago
Hi The code works fine on my side. Could you please provide more detail? Such as shape of score, value of K etc.
Hi The code works fine on my side. Could you please provide more detail? Such as shape of score, value of K etc.
Sry, it is caused by my mistake! Such a great work, many thanks!
Hi, I used these repo to transfer the following code from torch to trt.
However, the error shown here. [TensorRT] ERROR: (Unnamed Layer* 98) [TopK]: reduceAxes out of range
Are there some issues in convert/topk.py?