你们如何像keras here中那样可视化您的pytorch模型的训练历史。
我有一个受pytorch训练的模型，我想查看其训练图。 我可以只使用matplotlib吗？如果是的话，有人可以给我资源去追踪吗。
答案 0 :(得分：1)
您必须节省训练时的损失。训练有素的模型不会丢失。您需要再次训练。
保存训练时的损失，然后使用matplotlib将其相对于时期进行绘制。在您的训练功能中，正在计算损失的位置将其保存到文件中，然后可视化。 另外，如果要实时可视化，可以使用tensorboardX。
这是tensorboardX的教程：http://www.erogol.com/use-tensorboard-pytorch/