这里是我试错了多次之后,终于能够成功的方法
1.导入
from torch.utils.tensorboard import SummaryWriter
2.设置数据保存的路径
我这里的数据将会保存在,当前目录下的log文件夹之中(./表示当前文件夹)
writer = SummaryWriter('./log')
3. 画图需要的一些参数
ps:我这里是需要画出深度学习里面多次迭代下,cost数值的下降曲线
writer.add_scalar(tag=f'CNN-lr={lr}, epochs={epochs}', scalar_value=epoch_cost, global_step=e)
4.
writer.close()
5. 找到可视化的网址
我是使用pycharm的,所以,可以直接在pycharm里面菜单栏那里,找到View——Toolwindows——Terminal 打开终端
然后输入以下命令(注意:./log是我保存数据的文件夹路径),然后在输出中找到网址点击进去即可
tensorboard --logdir=./log
参考文章:PyTorch下的Tensorboard 使用 - 知乎 感谢大佬!