Tensorrt problem with ArgMax node

[ArtlyStyles]

Hi, I am studied this repo and it was very helpful. Now I am trying to convert the Lanenet (https://github.com/MaybeShewill-CV/lanenet-lane-detection) to tensorrt. I was able to conver the model and write the code. However, when I ran the inference, I got the following error:

[TRT] UffParser: Parser error: lanenet_model/vgg_backend/binary_seg/ArgMax: Reductions cannot be applied to the batch dimension.

Anyone knows why?