(self)
| 250 | self.last_iter = last_iter |
| 251 | |
| 252 | def _get_lr(self): |
| 253 | try: |
| 254 | pos = self.milestones.index(self.last_iter) |
| 255 | except ValueError: |
| 256 | return list(map(lambda group: group['lr'], self.optimizer.param_groups)) |
| 257 | except: |
| 258 | raise Exception('wtf?') |
| 259 | return list(map(lambda group: group['lr']*self.lr_mults[pos], self.optimizer.param_groups)) |
| 260 | |
| 261 | def get_lr(self): |
| 262 | return list(map(lambda group: group['lr'], self.optimizer.param_groups)) |