search

kuroko1t / nne

convert a pytorch model to a model for edge device
Apache License 2.0
18 stars 3 forks source link

add onnxsim #8

Closed kuroko1t closed 4 years ago

kuroko1t commented 4 years ago

Conversion of onnx model to TensorRT model fails. Due to Upsample layer, use onnx-simplifer as a countermeasure and delete unnecessary nodes.

https://github.com/NVIDIA/TensorRT/issues/284#issuecomment-568349214