hey @adaptivedecay, you can define your own scheduler for the learning rate by subclassing Trainer and overriding the create_scheduler function to include your logic: Trainer — transformers 4.5.0.dev0 documentation 3
alternatively, you can pass the optimizer and scheduler as a tuple in the optimizers argument.