G.Elham:

During training, data doesn’t become nan in get_item but after about 38 epochs trainloader returns tensors including nan values.


If you are using random transformations, could you check, if some of them might e.g. crop an invalid input? Are you able to reproduce the invalid values in the segmented image by using the DataLoader alone (without the model training) by just iterating it for 38 epochs?