(self)
| 108 | self.lr_decay = lr_decay |
| 109 | |
| 110 | def step(self): |
| 111 | for idx, p in enumerate(self.param_groups): |
| 112 | p['lr'] = self.init_lr_groups[idx] * (1 - self.curr_iter / self.max_iter) ** self.lr_decay |
| 113 | |
| 114 | |
| 115 | # just a try, not recommend to use |
no outgoing calls
no test coverage detected