【问题标题】:How to visualize my training history in pytorch?如何在 pytorch 中可视化我的训练历史?
【发布时间】:2019-08-02 10:08:24
【问题描述】:

你们如何在 keras here 中可视化 pytorch 模型的训练历史。

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

【问题讨论】:

    标签: matplotlib pytorch graph-visualization


    【解决方案1】:

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

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

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

    【讨论】:

    • 如何挽回损失?把它放在一个新的变量中?抱歉,我真的不知道。
    • dpaste.com/0T9A4GH 这是代码。最初我是从 github.com/paraschopra/one-network-many-uses 得到的代码。希望你能帮助我:)
    • 在第 49 行。将丢失写入文件。
    • 到文本文件还是放入变量?
    • 正如我对文本文件所说的
    猜你喜欢
    • 2019-12-20
    • 2020-11-01
    • 2017-04-24
    • 2020-05-06
    • 2020-02-03
    • 2018-10-12
    • 2017-11-14
    • 1970-01-01
    • 2018-10-13
    相关资源
    最近更新 更多