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