Skip to content

momo1986/GeodesicAdversarialTraining

Repository files navigation

GeodesicAdversarialTraining

  • 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)

About

An implementation of the work "Enhance Adversarial Robustness via Geodesic Distance"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages