Setting step_size in lr_scheduler.StepLR

1

can i set step size like step_size= [30,100,500] to specify in what epochs change the lr?

2

For multiple milestones you should use the lr_scheduler.MultiStepLR.

1 Like