Pytorch runtime error

1

Hi, when I run inference program, it warns me like that:
File “/usr/local/lib/python2.7/dist-packages/torch/nn/functional.py”, line 1045, in pad
raise NotImplementedError(“Only 4D and 5D padding is supported for now”)
NotImplementedError: Only 4D and 5D padding is supported for now

nn.ReflectionPad2d(3) was used in my program.
how can I resolve this problem

thank you!

2

What shape does your input have? It should be [batch, channels, width, height] for image data.
Could you check this?

3

problem has resolved, thank you!