Hi,
sorry, I didn’t adapt it well to your specific code. If just using predict in place of mask (based on accuracy involving predicted.eq(mask.data)) doesn’t work, what’s the shape of predict?
Best regards
Thomas