matplotlib - 如何在 pytorch 中可视化我的训练历史？

你们如何像在 keras 中一样可视化您的 pytorch 模型的训练历史。

我有一个经过 pytorch 训练的模型，我想看看它的训练图。我可以只使用matplotlib吗？如果是的话，有人可以给我资源来关注。

标签： matplotlibpytorchgraph-visualization

您必须在训练时保存损失。一个训练有素的模型不会有它的损失历史。你需要再次训练。

在训练时保存损失，然后使用 matplotlib 将其绘制到各个时期。在您的训练函数中，计算损失的地方将其保存到文件中并稍后将其可视化。此外，如果您想实时可视化，可以使用 tensorboardX。