Hello @futscdav, Thanks for the reply. I have taken K as 30 in there. And how computationally good is the batch training setup unlike mine accumulating loss sample by sample ? And is there a way to tackle the above problem and leverage GPU resources correctly ?