0

当我尝试从 github.com/behzadanksu/rl-attack 运行时

python train.py --env Breakout --noisy --save-dir ./data/Breakout/ --attack fgsm --num-steps 20 --attack-prob 1.0

从cleverhans 库中,我收到以下错误。我怎样才能避免它?

NotImplementedError:必须实现get_logits或必须定义 logits 输出fprop

4

1 回答 1

0

如果没有导致错误的原始代码片段,我的猜测是您使用的是在最初不是 CleverHans 的框架中定义的模型,并且您尝试将其包装到 CleverHans 模型中(这将有助于澄清您的问题)如果您可以发布一段代码定义你的模型)。如果是这种情况,那么您看到的错误意味着您包装原始模型的模型类没有方法get_logitsfprop已正确实现。您要么需要一个get_logits方法来接收输入张量并返回模型的输出张量,要么需要一个fprop方法来接收输入张量并填充字典,其中键设置为层名称和值到每个激活张量层。此字典的键之一必须是logitsprobs. model.py您可以在库文件中找到更多指导: https ://github.com/tensorflow/cleverhans/blob/master/cleverhans/model.py

于 2019-02-27T15:35:18.447 回答