Naveen kumar Jadi

How to create a TRT Model from ONNX or direct from PyTorch to TRT

14

hi @baudm ,@huyhoang17 actually im using this parseq to convert ONNX model like import torch parseq = torch.hub.load('baudm/parseq', 'parseq', pretrained=True, refine_iters=0).eval() dummy_input = torch.rand(1, 3, *parseq.hparams.img_size) # (1, 3, 32,...

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.

1

hi @qbxlvnf11 i wanted to convert parseq model to onnx , like below resnet18 = torch.hub.load('baudm/parseq', 'parseq', pretrained=True) this was added in main function resnet18.eval().to('cuda') this also added in the...

hi @qbxlvnf11 when i trying to convert onnx model to tensor rt (converted onnx from parseq model) import torch parseq = torch.hub.load('baudm/parseq', 'parseq', pretrained=True, refine_iters=0).eval() dummy_input = torch.rand(1, 3, *parseq.hparams.img_size)...