insightface
insightface copied to clipboard
Fixed the bug of max steps in lr scheduler while resuming the training
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.