(self)
| 244 | self.optimizer = optimizers.optim_entry(self.config.optimizer) |
| 245 | |
| 246 | def create_lr_scheduler(self): |
| 247 | if self.C.rank == 0: |
| 248 | self.logger.info('using new lr scheduler!') |
| 249 | self.config.lr_scheduler.kwargs.optimizer = self.optimizer |
| 250 | self.config.lr_scheduler.kwargs.last_iter = self.last_iter |
| 251 | self.config.lr_scheduler.kwargs.max_iter = self.config.max_iter |
| 252 | self.lr_scheduler = lr_scheduler_entry(self.config.lr_scheduler) |
| 253 | |
| 254 | def load(self, args): |
| 255 | if args.load_path == '': |
no test coverage detected