gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-04 17:39:40 27 4
gpt4 key购买 nike

你们如何像在 keras 中那样可视化您的 pytorch 模型的训练历史 here .

我有一个经过 pytorch 训练的模型,我想查看其训练图。我可以只使用 matplotlib 来做到这一点吗?如果是,有人可以给我资源以供引用。

最佳答案

你必须在训练时保存损失。训练有素的模型不会有损失的历史记录。你需要再次训练。

在训练时保存损失,然后使用 matplotlib 将其与历元相比较。在您的训练函数中,计算损失的地方将其保存到一个文件中,稍后再将其可视化。此外,如果您想实时可视化,可以使用 tensorboardX。

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

关于matplotlib - 如何在 pytorch 中可视化我的训练历史?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55112960/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com