| 9 | |
| 10 | |
| 11 | def get_args(): |
| 12 | parser = argparse.ArgumentParser() |
| 13 | utils.add_shared_args(parser) |
| 14 | |
| 15 | parser.add_argument('--perturb-freq', type=int, default=1, |
| 16 | help='set the perturbation frequency') |
| 17 | parser.add_argument('--report-freq', type=int, default=500, |
| 18 | help='set the report frequency') |
| 19 | parser.add_argument('--save-freq', type=int, default=5000, |
| 20 | help='set the checkpoint saving frequency') |
| 21 | |
| 22 | return parser.parse_args() |
| 23 | |
| 24 | |
| 25 | def regenerate_def_noise(def_noise, model, criterion, loader, defender, cpu): |