Open fan-chao opened 3 years ago
Hi,
When I used the script export_onnx.py to export .pt to onnx, some ScatterND layers are generated. However scatter_nd is not suppported for TensorRT.
I use the latest version yolov5-4.0 yolov5s.pt to exporting onnx. How do you handle this issue?
Thanks.
Hi,
When I used the script export_onnx.py to export .pt to onnx, some ScatterND layers are generated. However scatter_nd is not suppported for TensorRT.
I use the latest version yolov5-4.0 yolov5s.pt to exporting onnx. How do you handle this issue?
Thanks.