lr_scheduler#
Source code: tianshou/utils/lr_scheduler.py
- class MultipleLRSchedulers(*args: LRScheduler)[source]#
Bases:
object
A wrapper for multiple learning rate schedulers.
Every time
step()
is called, it calls the step() method of each of the schedulers that it contains. Example usage:scheduler1 = ConstantLR(opt1, factor=0.1, total_iters=2) scheduler2 = ExponentialLR(opt2, gamma=0.9) scheduler = MultipleLRSchedulers(scheduler1, scheduler2) policy = PPOPolicy(..., lr_scheduler=scheduler)