你能 plot 预训练的 model 的准确度图吗？ 深度学习

Question

我是深度学习的新手。 我完成了一个需要 8 小时运行的 model 的训练，但在关闭 jupyter 笔记本之前我忘记了 plot 准确度图。

我需要 plot 图表，我确实将 model 保存到我的硬盘上。 但是我如何 plot 预训练的 model 的准确度图？ 我在网上搜索了解决方案，结果一无所获。

任何帮助，将不胜感激！ 谢谢！

Answer 1

您使用了什么样的框架以及哪个版本？ 在未来您可能面临的问题中，这些信息可以在我们帮助您的方式中发挥关键作用。

不幸的是，对于 Pytorch/Tensorflow，您保存的 model 可能只保存神经元的权重，而不是其历史。 一旦 Jupyter Notebook 关闭，memory 就会被清理（以及你的训练历史数据）。

您唯一可以提取的是您拥有的最终损失/准确性。

但是，如果您定期保存 model 的版本，您可以加载它们并手动计算您需要的准确度/损失。 接下来，您可以使用 matplotlib 来重构图。

我知道这可能不是您想要的答案。 但是，如果硬件是你的，我建议你重新开始训练。 8h 在深度学习中训练 model 并不算多。