cleverhans - Cleverhans 加载预训练模型

Question

有没有办法加载预训练的模型。

我已经尝试了 load("model.joblib") 和 save("model.joblib", model) 函数，但是加载的模型在验证数据上只有大约 10% 的准确率，并且成功生成了大约 10% 的成功对抗样本。保存前模型准确率约为 99.3%，生成的对抗样本成功率约为 87%。如果我训练加载的加载模型与原始模型相同的时期数，那么我将获得预期的准确性和对抗性示例生成率。

有没有办法保存模型以便可以加载而不需要重新训练？

Answer 1

调用 load(filename) 后，您需要再次调用 model.get_logits(x) 和 CrossEntropy 以更新用于计算损失和准确度的值。