Hello I am still new in Semantic segmentation using pytorch. Suppose I have binary segmentation problem (where the class only 2). suppose I have size with this order [batch, channel, width,height]. I have batch size = 8.
 My output net = [8, 2, 96, 160] -> 2 in channel because have 2 class
 My ground truth(mask) = [8,1,96, 160] -> consist of 0 and 1 as the class

How can I measure the loss?, is there any way to use CrossEntrophy2D loss?