我正在使用别人创建的预训练模型，他们只是释放了模型权重。目前，我正在将模型权重导入到我的图形中，并通过张量名称来调用它们。但是，这似乎与cleverhans的代码不兼容，该代码似乎需要具有预测方法的模型对象。
有没有解决的办法，因为我没有模型类和预测方法，不需要我重写大部分的cleverhans攻击？
答案 0 :(得分：0)
您所描述的内容应该可行，但可能会占用大量资源，因为它可能会多次重新创建图形。基本上，您可以实现一个CleverHans模型类，该类在init方法中包含一个图形检查点。 get_logits或fprop方法应采用输入张量并通过执行某些图手术将检查点图的输入张量替换为您自己的张量来加载图以获得相应的输出张量：请参阅{{1 }}在`tf.import_graph_de中的参数：https://www.tensorflow.org/api_docs/python/tf/graph_util/import_graph_def