conda env create -f DUCAT.yaml
<DATASET>
: {cifar10
,cifar100
,tinyimagenet
}<ADV_TRAIN OPTION>
: {adv_train
,adv_trades
,adv_mart
}<CONSISTENCY_AT>
: {consistency
}
# Example for PGD-AT + DUCAT
python train.py --mode adv_train --dummy --epochs 130 --dataset cifar10 --augment_type base
# Example for MART + DUCAT
python train.py --mode adv_mart --dummy --epochs 130 --dataset cifar10 --augment_type base
# Example for Consistency-AT + DUCAT
python train.py --mode adv_train --consistency --dummy --epochs 130 --dataset cifar10 --augment_type base