| 135 | |
| 136 | |
| 137 | class StepLearningRateSchedule(LearningRateSchedule): |
| 138 | def __init__(self, initial, interval, factor): |
| 139 | self.initial = initial |
| 140 | self.interval = interval |
| 141 | self.factor = factor |
| 142 | |
| 143 | def get_learning_rate(self, epoch): |
| 144 | |
| 145 | return self.initial * (self.factor ** (epoch // self.interval)) |
| 146 | |
| 147 | |
| 148 | class WarmupLearningRateSchedule(LearningRateSchedule): |
no outgoing calls
no test coverage detected