├── .gitignore ├── Framework.PNG ├── LICENSE ├── OAT.py ├── PGDAT.py ├── README.md ├── attacks └── pgd.py ├── dataloaders ├── cifar10.py ├── stl10.py └── svhn.py ├── models ├── DualBN.py ├── FiLM.py ├── __init__.py ├── cifar10 │ ├── resnet.py │ ├── resnet_OAT.py │ ├── resnet_slimmable.py │ └── resnet_slimmable_OAT.py ├── slimmable_ops.py ├── stl10 │ ├── wide_resnet.py │ └── wide_resnet_OAT.py ├── svhn │ ├── wide_resnet.py │ └── wide_resnet_OAT.py └── wrap_cat_model.py └── utils ├── __init__.py ├── context.py ├── sample_lambda.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/.gitignore -------------------------------------------------------------------------------- /Framework.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/Framework.PNG -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/LICENSE -------------------------------------------------------------------------------- /OAT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/OAT.py -------------------------------------------------------------------------------- /PGDAT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/PGDAT.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/README.md -------------------------------------------------------------------------------- /attacks/pgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/attacks/pgd.py -------------------------------------------------------------------------------- /dataloaders/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/dataloaders/cifar10.py -------------------------------------------------------------------------------- /dataloaders/stl10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/dataloaders/stl10.py -------------------------------------------------------------------------------- /dataloaders/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/dataloaders/svhn.py -------------------------------------------------------------------------------- /models/DualBN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/DualBN.py -------------------------------------------------------------------------------- /models/FiLM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/FiLM.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/cifar10/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/cifar10/resnet.py -------------------------------------------------------------------------------- /models/cifar10/resnet_OAT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/cifar10/resnet_OAT.py -------------------------------------------------------------------------------- /models/cifar10/resnet_slimmable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/cifar10/resnet_slimmable.py -------------------------------------------------------------------------------- /models/cifar10/resnet_slimmable_OAT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/cifar10/resnet_slimmable_OAT.py -------------------------------------------------------------------------------- /models/slimmable_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/slimmable_ops.py -------------------------------------------------------------------------------- /models/stl10/wide_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/stl10/wide_resnet.py -------------------------------------------------------------------------------- /models/stl10/wide_resnet_OAT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/stl10/wide_resnet_OAT.py -------------------------------------------------------------------------------- /models/svhn/wide_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/svhn/wide_resnet.py -------------------------------------------------------------------------------- /models/svhn/wide_resnet_OAT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/svhn/wide_resnet_OAT.py -------------------------------------------------------------------------------- /models/wrap_cat_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/models/wrap_cat_model.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/utils/context.py -------------------------------------------------------------------------------- /utils/sample_lambda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/utils/sample_lambda.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Once-for-All-Adversarial-Training/HEAD/utils/utils.py --------------------------------------------------------------------------------