| 20 | |
| 21 | |
| 22 | def parse_args(): |
| 23 | parser = argparse.ArgumentParser("Submitit") |
| 24 | parser.add_argument("config_name", type=str, help="name of the config.") |
| 25 | parser.add_argument("--ngpus", default=None, type=int, help="Number of gpus to request on each node") |
| 26 | parser.add_argument("--nodes", default=None, type=int, help="Number of nodes to request") |
| 27 | parser.add_argument("--resume", default="", type=str, help="resume a checkpoint.") |
| 28 | parser.add_argument("--timeout", default=4320, type=int, help="Duration of the job") |
| 29 | parser.add_argument("--job_dir", default="", type=str, help="Job dir. Leave empty for automatic.") |
| 30 | parser.add_argument("--dependency", default=5, type=int, help="dependency for job failure.") |
| 31 | |
| 32 | parser.add_argument("--partition", default="learnlab", type=str, help="Partition where to submit") |
| 33 | parser.add_argument("--use_volta32", action='store_true', help="Request 32G V100 GPUs") |
| 34 | parser.add_argument('--comment', default="", type=str, help="Comment to pass to scheduler") |
| 35 | args = parser.parse_args() |
| 36 | return args |
| 37 | |
| 38 | |
| 39 | def get_shared_folder() -> Path: |