This repository implements Robust optimization with stylized pairing based on PyTorch API. The code was only validated on windows platform.
python stylized_pairing_train.py cifar --model resnet34
python stylized_pairing_test.py cifar --model resnet34 --type adamin
python plot_acc2perturbaton.py --dataset cifar --model resnet34