The first conv layer of your model (self.conv1) uses a single input channel, while you are providing input image tensors with 3 channels.
Set in_channels=3 in self.conv1 or transform the input to have a single channel only.
PS: it’s better to post code snippets by wrapping them into three backticks ```, as it makes debugging easier.