If you are passing the epoch argument to StepLR, it will be ste in this line of code 201 and used to calculate the learning rate here 100.
Since you are passing the average training loss, this should influence your learning rate scheduler.