- To run the Geodesic Adversarial Training
python GeodesicAT.py --net 'WRN'
python GeodesicAT.py --net 'resnet18'
- To run the evaluation program
python evaluate_attacks.py --model-path ${MODEL-PATH} --attack-method FGSM --dataset cifar10 (or cifar100)
An implementation of the work "Enhance Adversarial Robustness via Geodesic Distance"
python GeodesicAT.py --net 'WRN'
python GeodesicAT.py --net 'resnet18'
python evaluate_attacks.py --model-path ${MODEL-PATH} --attack-method FGSM --dataset cifar10 (or cifar100)
An implementation of the work "Enhance Adversarial Robustness via Geodesic Distance"