(self, start_lr, lr_power, total_iters)
| 11 | |
| 12 | class PolyLR(BaseLR): |
| 13 | def __init__(self, start_lr, lr_power, total_iters): |
| 14 | self.start_lr = start_lr |
| 15 | self.lr_power = lr_power |
| 16 | self.total_iters = total_iters + 0.0 |
| 17 | |
| 18 | def get_lr(self, cur_iter): |
| 19 | return self.start_lr * ((1 - float(cur_iter) / self.total_iters) ** self.lr_power) |
nothing calls this directly
no outgoing calls
no test coverage detected