@vfdev-5  Well, it would make sense if on 513th I was on epoch + 1.
But I don’t think that’s the case. At the completion of each epoch I log some metrics and this takes place on iteration 688 and not before. Hence, that’s when the epoch ends, right?