Closed andrewilyas closed 4 years ago
Ok so right off the bat a couple of things we should fix before we merge:
(resume_path is not None)
)~Ok the only immediately missing things before I can merge this in:
[x] I'm not fully convinced of the best_prec1
thing---could see some problems arising if someone is training adversarially and then wants to resume but with standard training. I think it should still be named either adv_prec1
or nat_prec1
(or something like that, and then we should just have a check for whether it's in the checkpoint).
[x] Need to update the docs for CLI training (everything else should be basically documented)
Probably enough for version 1.2 of the library?
random_perturb
for robust modelsrandom_step
(useful for training randomized smoothing models)with_image
option in AttackerModel)eps-fadein
argument in favor of a more general epsilon multiplier schedule)--resume-optimizer
(Fix #41)make_and_restore_model
now makes the model parallel even if resume_path is not specified, newadd_custom_forward
option for better interfacing with custom architectures