Quick question: is there a way to get the current_step number from the optimizer, or does the user have to explicitly keep count? I am assuming it would maintain a step count inside it because that value would be required in annealing the LR.
optimizer.state_dict() does give out a bunch of stuff but the current_step does not seem to be one of them.