Hi Peter, thanks for the response. Do you know any techniques I could use to overcome this issue ? I train some models but I have come up with overfitting. The validation loss and accuracy are saturating while training loss decreases and training accuracy gets to over 90%.
I tried to add weights to cross entropy loss and add a sampler when I define the train loader but neither of these methods work.
I am working with 435 DICOM files (160x384x384) and I have 5 classes.