Hi, it cannot be same because when I did the calculation, outputs is that:
train x size: torch.Size([77, 1, 16, 16, 3])
train y size:  torch.Size([77, 1, 16, 16])
val x size:  torch.Size([3465, 1, 3])
val y size:  torch.Size([3465, 1])