(self, args)
| 252 | self.lr_scheduler = lr_scheduler_entry(self.config.lr_scheduler) |
| 253 | |
| 254 | def load(self, args): |
| 255 | if args.load_path == '': |
| 256 | return |
| 257 | if args.recover: |
| 258 | # self.last_iter = load_state(args.load_path.replace('ckpt_task_', 'ckpt_task{}_'.format(self.ginfo.task_id)), self.model, optimizer=self.optimizer, recover=args.recover) |
| 259 | self.last_iter = load_state(args.load_path.replace('ckpt_task_', 'ckpt_task{}_'.format(self.ginfo.task_id)), self.model, optimizer=None, recover=args.recover) |
| 260 | self.last_iter -= 1 |
| 261 | else: |
| 262 | if args.load_single: |
| 263 | load_state(args.load_path, self.model, ignore=args.ignore) |
| 264 | else: |
| 265 | load_state(args.load_path.replace('ckpt_task_', 'ckpt_task{}_'.format(self.ginfo.task_id)), self.model, ignore=args.ignore) |
| 266 | |
| 267 | def initialize(self, args): |
| 268 | ## create dataset to get num_classes |
no test coverage detected