NVIDIA-AI-IOT / torch2trt

An easy to use PyTorch to TensorRT converter

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AttributeError when using torch.nn.functional.pad

clementabary opened this issue · comments

Hi all, when I run conversion of the following operation:

x_padded = torch.nn.functional.pad(x, (0, 0, pad_left, pad_right))

I get the error below:

AttributeError: 'tensorrt_bindings.tensorrt.INetworkDefinition' object has no attribute 'add_padding'. Did you mean: 'add_padding_nd'?

Do you know any way to circumvent this issue? Thanks in advance 🙏