Cleverhans加载预训练模型

Question

有没有办法加载一个预先训练好的模型。

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

有没有一种方法可以保存模型，这样就可以加载模型而不需要重新训练？

Answer 1

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