我完成了一个运行了 8 个小时的模型的训练,但我忘记在关闭 jupyter 笔记本之前绘制准确性图。
我需要绘制图表,并且我确实将模型保存到了我的硬盘上。但是如何绘制预训练模型的准确度图呢?我在网上搜索解决方案,但一无所获。
您使用的是什么类型的框架以及哪个版本?在您将来可能面临的问题中,这些信息可以在我们帮助您的方式中发挥关键作用。
不幸的是,对于 Pytorch/Tensorflow,您保存的模型可能会“仅保存神经元的权重”,而不是其历史记录。一旦 Jupyter Notebook 关闭,内存就会被清理(以及你的训练历史数据)。 您唯一可以提取的就是您的最终损失/准确性。
但是,如果您定期保存模型的版本,则可以加载它们并手动计算所需的准确性/损失。接下来,您可以使用 matplotlib 来重建图形。
我知道这可能不是您正在寻找的答案。但是,如果硬件是您的,我建议您重新开始训练。 8 小时对于训练深度学习模型来说并不算多。