(self)
| 56 | classification.main(self.args) |
| 57 | |
| 58 | def checkpoint(self): |
| 59 | import os |
| 60 | import submitit |
| 61 | |
| 62 | self.args.dist_url = get_init_file().as_uri() |
| 63 | checkpoint_file = os.path.join(self.args.output_dir, "checkpoint.pth") |
| 64 | if os.path.exists(checkpoint_file): |
| 65 | self.args.resume = checkpoint_file |
| 66 | print("Requeuing ", self.args) |
| 67 | empty_trainer = type(self)(self.args) |
| 68 | return submitit.helpers.DelayedSubmission(empty_trainer) |
| 69 | |
| 70 | def _setup_gpu_args(self): |
| 71 | import submitit |
no test coverage detected