@_joker …yes i am using both strong and weak augmentation on top of created Dataset object itself.Then used dataLoader for mini batch as part of my implmentation.I am using 50 epochs, at 10th epoch loss came down to 0.3 and after that it stopped decreasing.It starts fluctuating between 0.3 and 0.4