文件名称:consistency-adversarial:对抗性鲁棒性的一致性正则化
文件大小:534KB
文件格式:ZIP
更新时间:2024-03-28 22:11:20
data-augmentation adversarial-training consistency-regularization Python
对抗性鲁棒性的一致性正则化
, , , , 和进行的对抗性鲁棒一致性正则化的正式PyTorch实施。
1.依存关系
conda create -n con-adv python=3
conda activate con-adv
conda install pytorch torchvision cudatoolkit=11.0 -c pytorch
pip install git+https://github.com/fra31/auto-attack
pip install advertorch tensorboardX
2.培训
2.1。 培训选项和说明
培训方法的选项如下:
【文件预览】:
consistency-adversarial-master
----common()
--------train.py(2KB)
--------utils.py(795B)
--------__init__.py(0B)
--------eval.py(2KB)
--------common.py(4KB)
----figures()
--------augmentation.png(506KB)
----train.py(1KB)
----utils()
--------utils.py(5KB)
--------__init__.py(0B)
----models()
--------classifier.py(509B)
--------__init__.py(0B)
--------wide_resnet.py(4KB)
--------base_model.py(665B)
--------resnet.py(6KB)
----adv_lib()
--------__init__.py(0B)
--------mart.py(3KB)
--------attack.py(2KB)
--------trades.py(2KB)
----evals()
--------__init__.py(61B)
--------evals.py(3KB)
----datasets()
--------cifar_c.py(2KB)
--------datasets.py(4KB)
--------__init__.py(42B)
----eval.py(1KB)
----README.md(5KB)
----training()
--------train()
--------__init__.py(539B)
--------adv_train()