This is the official Pytorch implementation of CAS.
-
Train AT+CAS
python ./Standard_Adv_Training/train.py --gpu 0 --adv_train --affix AT_CAS
-
Train Trades+CAS
python ./Trades/train_trades_cifar10_cas.py
-
Train MART+CAS
python ./MART/train_mart_cas.py
Specify the checkpoint path as argument
python ./Trades/pgd_attack_cifar10_cas.py --model_path xxx