qbxlvnf11/convert-pytorch-onnx-tensorrt
Converting weights of Pytorch models to ONNX & TensorRT engines
PythonMIT
Issues
- 0
Should I add images from imagenet?
#4 opened by HyeonseopLim - 1
onnx和同仁堂结果不一致
#3 opened by henbucuoshanghai - 1
While Converting from PyTorch to ONNX Type Error: Type 'tensor(bool)' of input parameter (/Equal_27_output_0) of operator (CumSum) in node (/CumSum) is invalid.
#2 opened by naveenkumarkr723 - 0
In node 614 (parseGraph): INVALID_NODE: Invalid Node - /ArgMax std::bad_alloc
#1 opened by naveenkumarkr723