Python PyTorch ChainedScheduler用法及代码示例

本文简要介绍python语言中 torch.optim.lr_scheduler.ChainedScheduler 的用法。

用法:

class torch.optim.lr_scheduler.ChainedScheduler(schedulers)

参数：

schedulers(list) -链式调度程序列表。

学习率调度器的链表。它采用一系列可链接的学习率调度程序，并通过一次调用执行属于它们的连续 step() 函数。

示例

>>> # Assuming optimizer uses lr = 1. for all groups
>>> # lr = 0.09     if epoch == 0
>>> # lr = 0.081    if epoch == 1
>>> # lr = 0.729    if epoch == 2
>>> # lr = 0.6561   if epoch == 3
>>> # lr = 0.59049  if epoch >= 4
>>> scheduler1 = ConstantLR(self.opt, factor=0.1, total_iters=2)
>>> scheduler2 = ExponentialLR(self.opt, gamma=0.9)
>>> scheduler = ChainedScheduler([scheduler1, scheduler2])
>>> for epoch in range(100):
>>>     train(...)
>>>     validate(...)
>>>     scheduler.step()