(self, args)
| 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 |
| 269 | self.create_dataset() |
| 270 | self.create_model() |
| 271 | # self.create_optimizer() |
| 272 | ## load first to get last_iter |
| 273 | |
| 274 | # currently a workaround to get last_iter before sampler and scheduler |
| 275 | # if args.recover: |
| 276 | # self.last_iter = load_last_iter(args.load_path.replace('ckpt_task', 'ckpt_task{}'.format(self.ginfo.task_id))) |
| 277 | # self.last_iter -= 1 |
| 278 | self.create_optimizer() |
| 279 | self.load_args = args |
| 280 | |
| 281 | self.load(args) |
| 282 | ## then create sampler in dataloader |
| 283 | self.create_dataloader() |
| 284 | |
| 285 | self.create_lr_scheduler() |
| 286 | |
| 287 | def pre_run(self): |
| 288 | tmp = self.tmp |
nothing calls this directly
no test coverage detected