DeepXplore复现

复现了DeepXplore的 MNIST 部分,因为环境问题,对代码有所修改,以下是本人修改后的代码github传送门

主要内容

关于 TF 和 Keras 
1、不要用 tf 1.3 , 因为tf1.3对应的cuda会和自己的电脑显卡版本对不上,会报错,直接使用高一点版本的tf,例如 2.3
2、keras直接装最新版
    keras的引用全部改成 tensorflow.keras

环境配置
conda create --name DeepXplore  python=3.7
conda install cudatoolkit=10.1 cudnn=7.6.5
conda install tensorflow-gpu==2.3.0
conda install Keras
pip install Pillow
pip install h5py
pip install -i  https://pypi.tuna.tsinghua.edu.cn/simple  imageio
pip install  -i  https://pypi.tuna.tsinghua.edu.cn/simple   opencv-python


关于import
imsave:import imageio 
        相关代码从 imsave(...) 改成 imageio.imsave(...)
 to_categorical报错:   from tensorflow.python.keras.utils.np_utils import to_categorical

关于代码
gen_diff.py
    parser需要增加--:parser.add_argument('--xxx')
              需要把数据都设置 default,不然后面会有 NoneType
    loss 部分,- 单目运算符,需要加括号
              parser 部分未定义 default 会导致这里报错
    grads 部分,字典不能切片,要转换为list
            utils.py
                    layer_name, index = random.choice(list(model_layer_dict.keys()))

需要开 VPN

关于 Python 2.7 到 3.7 的问题

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值