(self, optimizer, multiplier, total_epoch, after_scheduler=None)
| 15 | """ |
| 16 | |
| 17 | def __init__(self, optimizer, multiplier, total_epoch, after_scheduler=None): |
| 18 | self.multiplier = multiplier |
| 19 | if self.multiplier < 1.: |
| 20 | raise ValueError('multiplier should be greater thant or equal to 1.') |
| 21 | self.total_epoch = total_epoch |
| 22 | self.after_scheduler = after_scheduler |
| 23 | self.finished = False |
| 24 | super(GradualWarmupScheduler, self).__init__(optimizer) |
| 25 | |
| 26 | def get_lr(self): |
| 27 | if self.last_epoch > self.total_epoch: |
nothing calls this directly
no outgoing calls
no test coverage detected