还有另一种方法可以保存图的保存损失和学习率

时间:2019-11-30 07:54:27

标签: pytorch fast-ai

我使用fastai的学习课程在远程云服务器上进行培训。 通常,我们会根据学习率与损失之间的关系来调整学习率。

learn.lr_find()

learn.recorder.plot()

但是,由于某种原因,远程服务器无法返回此关系图。

还有其他方法可以保存它们之间的关系并在本地绘制吗?

1 个答案:

答案 0 :(得分:0)

好的,我找到了答案。

graph=learn.recorder.plot(return_fig=True)
graph.savefig('loss.png')

它将返回一个名为loss.png的图像。

我可以通过这张图片查看损失与学习率之间的关系