DeepRobust使用
下载
github地址:https://github.com/DSE-MSU/DeepRobust
论文链接:https://arxiv.org/abs/2005.06149
安装
pip install deeprobust
注意事项
scipy 版本 1.3.1 如果用pip直接安装 版本错误
2.使用CPU
需要修改,base_attack.py和使用的算法中
def init(self, model, device = ‘cpu’): # change ‘cuda’
self.model = model
self.device = device
3.
数据集下载,如果使用网络下载太慢,可以先下载到本地,然后使用
下载地址:https://www.cs.toronto.edu/~kriz/cifar.html
修改文件
base_folder = ‘cifar-10-batches-py’
# url = “https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz”
url