(self, current_step)
| 39 | return lr_scheduler.LambdaLR(optimizer, self.lambda_func, self.last_epoch) |
| 40 | |
| 41 | def lambda_func(self, current_step): |
| 42 | if current_step < self.warmup_epoch: |
| 43 | return float(current_step) / float(max(1, self.warmup_epoch)) |
| 44 | return self.gamma ** (current_step // self.step_size) |
| 45 | |
| 46 | |
| 47 | class MultiStepLR(object): |
nothing calls this directly
no outgoing calls
no test coverage detected