| 12 | |
| 13 | |
| 14 | def parse_args(): |
| 15 | detection_parser = detection.get_args_parser() |
| 16 | parser = argparse.ArgumentParser("Submitit for detection", parents=[detection_parser]) |
| 17 | parser.add_argument("--ngpus", default=8, type=int, help="Number of gpus to request on each node") |
| 18 | parser.add_argument("--nodes", default=1, type=int, help="Number of nodes to request") |
| 19 | parser.add_argument("--timeout", default=60, type=int, help="Duration of the job") |
| 20 | parser.add_argument("--cpus_per_task", default=16, type=int, help="Duration of the job") |
| 21 | parser.add_argument("--job_dir", default="", type=str, help="Job dir. Leave empty for automatic.") |
| 22 | parser.add_argument("--job_name", type=str, help="Job name.") |
| 23 | parser.add_argument("--qos", type=str, default=None, help="specify preemptive QOS.") |
| 24 | parser.add_argument("--requeue", action='store_true', help="job requeue if preempted.") |
| 25 | parser.add_argument("--mail_type", type=str, default='ALL', help=" send email when job begins, ends, fails or preempted.") |
| 26 | parser.add_argument("--mail_user", type=str, default='', help=" email address.") |
| 27 | # refer to https://slurm.schedmd.com/sbatch.html & \ |
| 28 | # https://github.com/facebookincubator/submitit/blob/11d8f87f785669e8a01aa9773a107f9180a63b09/submitit/slurm/slurm.py \ |
| 29 | # for more details about parameters of slurm. |
| 30 | return parser.parse_args() |
| 31 | |
| 32 | |
| 33 | def get_shared_folder() -> Path: |