【文件属性】:
文件名称:AWP:NeurIPS 2020论文代码“专家权重摄动有助于稳健概括”
文件大小:380KB
文件格式:ZIP
更新时间:2021-05-10 09:23:08
Python
对抗性重量扰动有助于稳健概括
代码NeurIPS 2020“”的,,和。
消息
2020年10月13日-我们的代码和文件已发布。
必要的
该代码是在PyTorch中实现的,我们已经在以下环境设置下测试了该代码:
Python= 3.7.3
火炬= 1.2.0
火炬视觉= 0.4.0
该存储库中有什么
基于AWP的对抗训练(AT-AWP)的代码位于at-awp ,而基于AWP的./trades-awp -AWP)的代码位于./trades-awp :
在./at-awp ,CIFAR-10,CIFAR-100和SVHN的代码分别在train_cifar10.py , train_cifar100.py和train_svhn.py 。
在./trades-awp ,CIFAR-10和CIFAR-100的代码在train_trades_cifar.py 。
可以在或(pw:8tsv)
【文件预览】:
AWP-main
----.gitattributes(66B)
----trades_AWP()
--------utils()
--------models()
--------utils_awp.py(2KB)
--------train_trades_cifar.py(13KB)
----auto_attacks()
--------logs()
--------eval.py(5KB)
--------README.md(7KB)
----cover.png(229KB)
----LICENSE(1KB)
----AT_AWP()
--------train_cifar100.py(14KB)
--------logs()
--------wideresnet.py(4KB)
--------awp.py(2KB)
--------utils.py(4KB)
--------preactresnet.py(4KB)
--------train_cifar10.py(20KB)
--------utils_awp.py(2KB)
--------train_svhn.py(15KB)
----README.md(5KB)