When to call scheduler.step?

[benihime91]

scheduler.step.is call after each batch or epoch?

[saniazahan]

indeed, I am also curious as we usually call optim.step() after each batch