Hmm… not easily, unfortunately.   I’m using a large custom dataset for segmentation with SGD and BCE + Dice so nothing too crazy there. The model I’m using is here 3  I’ll try another model just to see if that’s where the issue lies.