scaled-yolov4-large model failed

talebolano / TensorRT-Scaled-YOLOv4

TensorRT for Scaled YOLOv4(yolov4-csp.cfg)

10 stars 4 forks source link

Open KevenLee opened 2 years ago

KevenLee commented 2 years ago

Hi， when I trans. yolov4-p6 model from pytorch pth format to onnx, the error happen.

ModuleNotFoundError: No module named 'mish_cuda'

KevenLee commented 2 years ago

If I use mish-cuda, the onnx reasoning of the p6 model is very different from the pytorch pt reasoning.