将预先训练的resnet模型加载为cleverhans模型格式

时间:2019-07-30 07:08:58

标签: python tensorflow resnet cleverhans

我正在尝试将来自MadryLab CIFAR-10挑战的预先训练好的ResNet模型加载到CleverHans中,以计算传输攻击。

但是,无法将保存的模型恢复到model_zoo.madry_lab_challenges.cifar10_model.ResNet对象中。最初似乎可以恢复正常,但是当我尝试实际使用模型时,出现如下错误:

  

尝试使用未初始化的值   ResNet / unit_3_1 / residual_only_activation / BatchNorm / moving_mean

重现此错误的最简单方法是实际上只运行attack_model.py中提供的CleverHans示例:

https://github.com/tensorflow/cleverhans/blob/master/examples/madry_lab_challenges/cifar10/attack_model.py

在尝试使用模型时,它在adv_trainednaturally_trained上都遇到相同的错误。

是否有解决此问题的方法?

似乎另一个选择是改为使用cleverhans.model.CallableModelWrapper,但是我还没有找到如何使用它的示例。

0 个答案:

没有答案