No actually it’s a semantic segmentation network. So the “labels” are for each pixel. There are only two possible classes for the pixels. Each image is 2560x256.
All my doubts started because I’ve just changed GPU, now I have a RTX, I was trying to use FP16 training that should enable tensor cores (with apex) but I have absolutely no benefit (actually it is almost slower…). I was trying to understand if this part that is constrained not to use FP16 may be the problem.