The first conv layer expects input tensors with a single channel, while your inputs contain 3 channels.
You would thus either have to change the first layer to:
nn.Conv2d(3, 32, 3, padding=1)

or use grayscale images.