Train with a PyTorch optimizer and StepLR schedule, recording the learning rate applied at each epoch.
