I am facing one more problem… I want to use cross-entropy for semantic segmentation and the dimension of my target [4,3,200,200] and output [4,8,200,200]. giving me an error. 8 represent the number of classes for classification. Because if I am using the same network with output_dim =1 for MSE it is giving me the output. but for semantic, I am getting errors.
Screenshot from 2019-06-12 15-59-10.png1920×1080 754 KB