我正在使用其他人创建的预训练模型，他们只发布了模型权重。目前我正在将模型权重导入我的图表并通过张量名称调用它们。但是，这似乎与cleverhans 的代码不兼容，该代码似乎需要具有方法predict 的模型对象。
是否有任何解决方法不需要我重写大多数cleverhans 攻击，因为我没有模型类和预测方法？
问问题
101 次
1 回答
0
您所描述的内容应该是可能的，但可能会占用资源，因为它可能会多次重新创建图表。基本上，您可以实现一个 CleverHans 模型类，该类在 init 方法中接受一个图形检查点。get_logitsorfprop方法应该接受一个输入张量并加载图以通过执行一些图手术以用您自己的张量替换检查点图的输入张量来获得相应的输出张量：请参阅 `tf.import_graph_de 中的参数input_map：https://www。 tensorflow.org/api_docs/python/tf/graph_util/import_graph_def
于 2019-02-27T15:41:17.653 回答