Cleverhans 加载预训练模型

Cleverhans load pre-trained model

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

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

有没有办法保存模型,这样就可以加载它而不需要重新训练?

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