Hey so here’s the sizes of the variables:
inp getting passed in [1, 10, 256, 256] -> batch size 1, 10 classes, 256 x 256 image
target [256, 256]
mask [256, 256]
log_p after the masking operation: [28585, 10]
target after the masking operation: [28585]
My target contains class values between 0 and 9 for each pixel.
I’m confused when you say to pass the original output and target with a non-reducing criterion then zero out the mask. Isn’t the output of the loss function is a scalar value, how would I zero the mask?