你们如何像在Keras这里一样，将PyTorch模型的训练历史可视化呢？
我有一个已经训练好的PyTorch模型，我想查看其训练图。是否可以只使用Matplotlib完成此操作？如果可以，有没有资源可以提供给我参考。
1个回答
1
您需要在训练过程中保存损失。训练完成后的模型不会保留其损失历史记录。您需要重新进行训练。
在训练函数中，计算损失时，请将其保存到文件中，并使用matplotlib将其与纪元一起绘制出来。如果想要实时可视化，也可以使用tensorboardX。
这是有关tensorboardX的教程：http://www.erogol.com/use-tensorboard-pytorch/
在训练函数中，计算损失时，请将其保存到文件中，并使用matplotlib将其与纪元一起绘制出来。如果想要实时可视化，也可以使用tensorboardX。
这是有关tensorboardX的教程：http://www.erogol.com/use-tensorboard-pytorch/
- Abhishek Verma
5
1我该如何保存损失？将其放入一个新变量中吗？很抱歉，我真的不知道。 - user11140802
这是代码的链接：http://dpaste.com/0T9A4GH。最初我从https://github.com/paraschopra/one-network-many-uses获取了该代码。希望能帮到您 :) - user11140802
在第49行写入损失到文件中。 - Abhishek Verma
将文本写入文本文件还是放在变量中？ - user11140802
如我所说，将其写入一个文本文件中。 - Abhishek Verma
网页内容由stack overflow 提供, 点击上面的可以查看英文原文，
原文链接
原文链接