pytorch训练过程可视化

1、本机训练可视化

如果runs是生成log的目录,执行下面的命令就可以查看训练的过程:

tensorboard --logdir runs

在这里插入图片描述

从结果可以看出,loss并不是稳定下降的完美曲线,所以可能需要调整某些参数。

参考:Pytorch使用tensorboardX可视化。超详细!!!

2、服务器训练 可视化

在远程服务器上执行如下命令启动tensorboard的服务之后:

tensorboard --logdir logs

无法在本地PC上打开tensorboard的网页:

http:

猜你喜欢

转载自blog.csdn.net/juluwangriyue/article/details/120799678
今日推荐