(self, optimizer, total_iters, power=1.0, last_epoch=-1)
| 101 | |
| 102 | class PolynomialLR(_LRScheduler): |
| 103 | def __init__(self, optimizer, total_iters, power=1.0, last_epoch=-1): |
| 104 | self.total_iters = total_iters |
| 105 | self.power = power |
| 106 | super().__init__(optimizer, last_epoch) |
| 107 | |
| 108 | def get_lr(self): |
| 109 | if self.last_epoch == 0 or self.last_epoch > self.total_iters: |