(self, args)
| 136 | self.model = model |
| 137 | |
| 138 | def load(self, args): |
| 139 | if args.load_path == '': |
| 140 | return |
| 141 | if args.recover: |
| 142 | self.last_iter = load_state(args.load_path.replace('ckpt_task_', 'ckpt_task{}_'.format(\ |
| 143 | self.ginfo.task_id)), self.model, recover=args.recover) |
| 144 | self.last_iter -= 1 |
| 145 | else: |
| 146 | if args.load_single: |
| 147 | load_state(args.load_path, self.model, ignore=args.ignore) |
| 148 | else: |
| 149 | load_state(args.load_path.replace('ckpt_task_', 'ckpt_task{}_'.format(\ |
| 150 | self.ginfo.task_id)), self.model, ignore=args.ignore) |
| 151 | |
| 152 | def initialize(self, args): |
| 153 |
no test coverage detected