RuntimeError: Expected 5-dimensional input for 5-dimensional weight [32, 3, 1, 5, 5], but got 4-dimensional input of size [3, 256, 128, 128] instead
Based on the error message it seems you are using an nn.Conv3d module, which expect a 5-dimensional tensor as the input in the shape [batch_size, channels, depth, height, width], while one of these dimensions are missing (given the posted shape I guess it could be the channel dimension).
thank you yes i think so i will check