What is the proper way of using last_epoch in a lr_scheduler?

Asked the new question concerning the out of memory issue in 0.4 :