@shivam2298 I am not fully understand what you are saying.
Will it not lead to size mismatch.
As prediction-
10* 2 *128 *128
And target-
10* 128 *128
if its about ,that we need to convert our target to one hot. So Please explain it, how i can convert it in 2 channel dimension with only 1 and 0 as labels.