Oh, that function accepts a list for each parameter group. I tested it, and it works fine.

image901×518 30.8 KB

Just one note that when you use OneCycleLR, the learning rate in the optimizer does not matter. What you put in max_lr determines your learning rate!