将预训练resnet模型加载到cleverhans模型表单

2024-10-04 05:25:19 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图将预先训练好的ResNet模型从MadryLab CIFAR-10挑战加载到CleverHans中,以计算传输攻击。你知道吗

但是,将保存的模型恢复到model_zoo.madry_lab_challenges.cifar10_model.ResNet对象中是行不通的。一开始似乎恢复得很好,但当我尝试实际使用模型时,会出现如下错误:

Attempting to use uninitialized value ResNet/unit_3_1/residual_only_activation/BatchNorm/moving_mean

重现此错误的最简单方法是实际运行包含在CleverHans中的attack_model.py示例,如下所示:

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

在加载模型后,当它试图在adv_trainednaturally_trained模型上使用它时,它会遇到相同的错误。你知道吗

这个问题有解决办法吗?你知道吗

另一种选择似乎是使用cleverhans.model.CallableModelWrapper,但是我还没有找到一个如何使用它的例子。你知道吗


Tags: py模型model错误labresnetchallengesattack