你们如何形象化你们的pytorch模型的训练历史,如keras here。
我有一个pytorch训练的模型,我想看到它的训练图。我可以只使用matplotlib吗?如果是的话,有人可以给我资源。
你必须在训练时节省损失。训练有素的模型不会有丢失的历史。你需要再次训练。
在训练时节省损失,然后使用matplotlib将其绘制在时代周围。在您的训练函数中,计算损失时将其保存到文件中并在以后进行可视化。此外,如果要实时可视化,可以使用tensorboardX。
这是tensorboardX:http://www.erogol.com/use-tensorboard-pytorch/的教程