Hi, thanks for the reply! I turned off the “shuffle” parameter for my dataloader, which I think should now guarantee that each epoch should go through all of the training images sequentially. So far, after a couple epochs of training, no error. What I figured was that if it was a mask image somewhere in the training set causing the issue, having the dataloader go through all of them SHOULD force the error to pop up, right?