我正在尝试将来自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
示例:
在尝试使用模型时,它在adv_trained
和naturally_trained
上都遇到相同的错误。
是否有解决此问题的方法?
似乎另一个选择是改为使用cleverhans.model.CallableModelWrapper
,但是我还没有找到如何使用它的示例。