In node 614 (parseGraph): INVALID_NODE: Invalid Node - /ArgMax std::bad_alloc

[naveenkumarkr723]

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) # (1, 3, 32, 128) by default

############ To ONNX
parseq.to_onnx('parseq.onnx', dummy_input, opset_version=14) #2 opset v14 or newer is required

this onnx i passed for the TRT conversion but getting the below message and not able to convert trt model

Succeeded finding ONNX file!
Failed parsing .onnx file!
In node 614 (parseGraph): INVALID_NODE: Invalid Node - /ArgMax
std::bad_alloc

command used to convert onnx to trt

python convert_onnx_to_tensorrt.py --dynamic_axes True --sample_image_path roi_1.jpg --onnx_model_path parseq_ref_simple.onnx --tensorrt_engine_path FP32_explicit.engine --engine_precision FP32 2>&1| tee logs.log