Required Packages:

torch == 1.7.1
torchvision == 0.8.2
numpy == 1.21.1
sklearn == 0.24.2
matplotlib == 3.4.2
tqdm == 4.62.0

Running Commands:

#Backdoor attacks

CUDA_VISIBLE_DEVICES=0 python train_poison_cifar.py --max_epoch=120 --poison-rate=0.005 --transparency=0.4 --use_trigger='cube' --use_clean_label   --adv_train --epsilon=0.00784 --alpha=0.00196 --y-target=2


#PGD defense

CUDA_VISIBLE_DEVICES=0 python PGD_detection.py --file_name=XXXX  --poison-rate=0.015 --transparency=1.0 --use_trigger=cube