| 143 | |
| 144 | |
| 145 | def parse_args(): |
| 146 | parser = argparse.ArgumentParser(description='Dispatcher to run all experiments') |
| 147 | |
| 148 | parser.add_argument("--gpu", type=str, default='0,1,2,3', |
| 149 | help='list of available gpus') |
| 150 | parser.add_argument("--n_gpu", type=int, default=1, |
| 151 | help="number of gpus per job") |
| 152 | parser.add_argument("--model", type=str, default='blocklm-roberta-large') |
| 153 | parser.add_argument("--task", type=str, required=True) |
| 154 | parser.add_argument("--overwrite", action='store_true', default=False, |
| 155 | help='whether to rerun experiments with the same result ' |
| 156 | 'file location') |
| 157 | parser.add_argument("--debug", action='store_true', default=False) |
| 158 | |
| 159 | return parser.parse_args() |
| 160 | |
| 161 | |
| 162 | def main(): |