python - 将预训练的 resnet 模型加载到cleverhans 模型格式

Question

我正在尝试将预训练的 ResNet 模型从MadryLab CIFAR-10挑战加载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_trained和naturally_trained模型上使用它时遇到相同的错误。

这个问题有解决方法吗？

似乎另一种选择是使用cleverhans.model.CallableModelWrapper，但我无法找到如何使用它的示例。