Solved by ptrblck in post #19 
              
              
                Thanks for mentioning it again. 
The dice loss is used for discrete data, so I doubt it can work for a regression task. 
Since your labels are in the range [0, 1], I would again recommend to use nn.BCEWithLogitsLoss. 
Alternatively, since you are already rounding the targets to 0 and 1 to be able t…