thank you. now my targets are (16, 256, 256) and the predictions are (16, 12, 256, 256). can i apply cross entropy loss function on these two data? no need for same dimension ?
edit: is there no need for one hot encoding the labels? can i directly feed the single channel mask (convert color to grayscale) to cross entropy loss function?