tensorRT
tensorRT copied to clipboard
yolov5 run error:Found unsupported datatype (11) when importing initializer: model.0.conv.total_ops
hello,I follow the Quick Start,and get yolov5s.onnx from https://github.com/Syencil/mobile-yolov5-pruning-distillation, when run the ./bin/yolov5 raise the error "Found unsupported datatype (11) when importing initializer: model.0.conv.total_ops". I check the onnx export log, "DOUBLE" datatype exists, may be this lead to error, how can i solve this, any help will be appreciated. I use the pytorch1.4 and TensorRT7.1.3.4.