如何在pytorch中可视化我的训练历史?

问题描述 投票:0回答:1

你们如何形象化你们的pytorch模型的训练历史,如keras here

我有一个pytorch训练的模型,我想看到它的训练图。我可以只使用matplotlib吗?如果是的话,有人可以给我资源。

matplotlib pytorch graph-visualization
1个回答
1
投票

你必须在训练时节省损失。训练有素的模型不会有丢失的历史。你需要再次训练。

在训练时节省损失,然后使用matplotlib将其绘制在时代周围。在您的训练函数中,计算损失时将其保存到文件中并在以后进行可视化。此外,如果要实时可视化,可以使用tensorboardX。

这是tensorboardX:http://www.erogol.com/use-tensorboard-pytorch/的教程

© www.soinside.com 2019 - 2024. All rights reserved.