nn.Conv2d expects an input in the shape
[batch_size, channels, height, width] while you are passing a 5-dimensional input as given in the error message.
Remove the unnecessary dimension and
permute the input to have the aforementioned channels-first memory layout.
PS: you can post code snippets by wrapping them into three backticks ```.