(self, current_step)
| 80 | return lr_scheduler.LambdaLR(optimizer, self.lambda_func, self.last_epoch) |
| 81 | |
| 82 | def lambda_func(self, current_step): |
| 83 | if current_step < self.warmup_epoch: |
| 84 | return float(current_step) / float(max(1.0, self.warmup_epoch)) |
| 85 | return 1.0 |
| 86 | |
| 87 | |
| 88 | class LinearLR(object): |
nothing calls this directly
no outgoing calls
no test coverage detected