| 21 | |
| 22 | |
| 23 | def arg_parser(): |
| 24 | parser = argparse.ArgumentParser() |
| 25 | parser.add_argument("--patch_size", default = 256, type = int) |
| 26 | parser.add_argument("--batch_size", default = 16, type = int) |
| 27 | parser.add_argument("--total_iter", default = 50000, type = int) |
| 28 | parser.add_argument("--adv_train_lr", default = 2e-4, type = float) |
| 29 | parser.add_argument("--gpu_fraction", default = 0.5, type = float) |
| 30 | parser.add_argument("--save_dir", default = 'pretrain') |
| 31 | |
| 32 | args = parser.parse_args() |
| 33 | |
| 34 | return args |
| 35 | |
| 36 | |
| 37 | |