【发布时间】:2019-08-02 10:08:24
【问题描述】:
你们如何在 keras here 中可视化 pytorch 模型的训练历史。
我有一个经过 pytorch 训练的模型,我想查看它的训练图。 我可以只使用matplotlib吗?如果是的话,有人可以给我资源吗?
【问题讨论】:
标签: matplotlib pytorch graph-visualization
你们如何在 keras here 中可视化 pytorch 模型的训练历史。
我有一个经过 pytorch 训练的模型,我想查看它的训练图。 我可以只使用matplotlib吗?如果是的话,有人可以给我资源吗?
【问题讨论】:
标签: matplotlib pytorch graph-visualization
您必须在训练时保存损失。一个训练有素的模型不会有它的损失历史。你需要重新训练。
在训练时保存损失,然后使用 matplotlib 将其绘制到各个时期。在您的训练函数中,正在计算损失的地方将其保存到文件中并稍后将其可视化。 此外,如果您想实时可视化,可以使用 tensorboardX。
这是一个tensorboardX的教程:http://www.erogol.com/use-tensorboard-pytorch/
【讨论】: