don’t reset grad.step