Fix Model Saving and Printed Message in Evaluation

s-huu / TurningWeaknessIntoStrength

Official implementation for paper: A New Defense Against Adversarial Images: Turning a Weakness into a Strength

37 stars 10 forks source link

Closed Xiuyu-Li closed 4 years ago

Xiuyu-Li commented 4 years ago

Save the best model as model_best.pth.tar in train_vgg19.py, which is consistent with the model name used in attack.py and detect.py.
Fix the bug of printing the number of successful adversary when doing evaluation.