def _conv_forward(self, input: Tensor, weight: Tensor, bias: Optional[Tensor]):
 if self.padding_mode != "zeros":
 return F.conv3d(
 F.pad(
 input, self._reversed_padding_repeated_twice, mode=self.padding_mode
 ),
 weight,
 bias,
 self.stride,
 _triple(0),
 self.dilation,
 self.groups,
 )
> return F.conv3d(
 input, weight, bias, self.stride, self.padding, self.dilation, self.groups
 )
E RuntimeError: Given groups=1, weight of size [160, 12, 3, 3, 3], expected input[1, 3, 3, 258, 258] to have 12 channels, but got 3 channels instead
/opt/conda/lib/python3.11/site-packages/torch/nn/modules/conv.py:720: RuntimeError