Linearly increases the learning rate between two boundaries over a number of iterations.
| 39 | |
| 40 | |
| 41 | class LinearLR(_LRSchedulerMONAI): |
| 42 | """Linearly increases the learning rate between two boundaries over a number of |
| 43 | iterations. |
| 44 | """ |
| 45 | |
| 46 | def get_lr(self): |
| 47 | r = self.last_epoch / (self.num_iter - 1) |
| 48 | return [base_lr + r * (self.end_lr - base_lr) for base_lr in self.base_lrs] |
| 49 | |
| 50 | |
| 51 | class ExponentialLR(_LRSchedulerMONAI): |
no outgoing calls
no test coverage detected
searching dependent graphs…