Cleverhans负荷预训练模型

时间:2020-05-11 21:35:54

标签: cleverhans

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

我尝试了load(“ model.joblib”)和save(“ model.joblib”,model)函数,但是加载的模型在验证数据上的准确度仅为10%,并且成功生成了大约10%的成功对抗性例子。 在保存之前,模型的准确性约为99.3%,并且大约有87%的对抗性示例获得了成功。 如果以与原始数相同的纪元数训练加载的模型,则可以得到预期的准确性和对抗性示例生成率。

是否有一种方法可以保存模型,以便无需重新训练即可加载模型?

1 个答案:

答案 0 :(得分:0)

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