(self, lr_policy)
| 83 | return decayed_lr |
| 84 | |
| 85 | def get_lr(self, lr_policy): |
| 86 | if lr_policy.lower() == 'poly': |
| 87 | decayed_lr = self.poly_decay() |
| 88 | elif lr_policy.lower() == 'piecewise': |
| 89 | decayed_lr = self.piecewise_decay() |
| 90 | elif lr_policy.lower() == 'cosine': |
| 91 | decayed_lr = self.cosine_decay() |
| 92 | else: |
| 93 | raise Exception( |
| 94 | "unsupport learning decay policy! only support poly,piecewise,cosine" |
| 95 | ) |
| 96 | |
| 97 | decayed_lr = self.lr_warmup(decayed_lr, 0, cfg.SOLVER.LR) |
| 98 | return decayed_lr |
| 99 | |
| 100 | def sgd_optimizer(self, lr_policy, loss): |
| 101 | decayed_lr = self.get_lr(lr_policy) |
no test coverage detected