Hey. thanks for your reply. So i followed your suggestion and my ground truth contains pixel with 0 and 255. i got an error in criterion steps

ValueError: Target size (torch.Size([1, 224, 224])) must be the same as input size (torch.Size([1, 2, 224, 224]))


does that mean that i have to transform my input image to Grayscale first ? because i notice the first dimension of RGB (3) became (2) in training steps .