上传者: 42104181
|
上传时间: 2021-12-08 11:36:01
|
文件大小: 1.23MB
|
文件类型: -
使用FGSM进行快速对抗训练
使用FGSM对手实施快速对抗训练代码的存储库,能够在6分钟内训练强大的CIFAR10分类器,并在12小时内训练强大的ImageNet分类器。 由 , 和。 请参阅我们的arXiv上的论文,这是由自由对抗训练纸灵感通过Shafahi等。 (2019)。
消息
12/19/2019-ICLR 2020接受
1/14/2019-arXiv发布并发布存储库
该存储库中有什么?
带有针对MNIST,CIFAR10和ImageNet的随机初始化的FGSM对抗训练方法的实现
使用库进行和混合精度训练,以实现类似DAWNBench的加速
使用此代码库的预训练模型
ImageNet代码大部分是从派生的,并进行了相应的修改,以进行快速FGSM对抗训练
安装及使用
所有示例都可以在不使用PyTorch v1.0或更高版本进行混合精度的情况下运行
要使用混合精度培训,请按照的顶点安