I am going through the transfer learning tutorial, and looked at the source code of StepLR.
It seems if I wanted to change the StepLR to be based on batches instead of epochs,
I just need to run scheduler.step() every x of amount batches, where x is the amount desired and set step_size within lr_scheduler to 1?