Issues with torch.nn.ReflectionPad2d(padding) conversion to TRT engine

Can anyone please let me know if this issue was solved or not? Thanks in advance!

Hi @ci20l ,

We could generate TRT engine successfully on latest TensorRT version.
I think padding related node is causing error, we don’t support 2D shape tensors yet. We can try workaround constant-fold with polygraphy. After this we are able to successfully generate engine. Please try
polygraphy surgeon sanitize --fold-constants fmnist_cnn.onnx -o folded.onnx

For more details,

Thank you.