insightface icon indicating copy to clipboard operation
insightface copied to clipboard

Fixed the bug of max steps in lr scheduler while resuming the training

Open vinits5 opened this issue 2 years ago • 0 comments

Let's say you started the training for 5 epochs and killed/stopped it at 4th epoch. While resuming, you updated the total epochs to 10 then, the max_steps variable in lr_scheduler needs to be updated as per the new total epochs. This change enables it.

vinits5 avatar Oct 19 '22 06:10 vinits5