文章摘自详解PyTorch可视化工具visdom(一)-CSDN博客
模型训练过程中需要实时监听并可视化一些数据,如损失值loss,正确率acc等。在tensorflow中,使用的工具为tensorboard;
安装一下试试
1.安装
pip install visdom
2.启动
python -m visdom.server
在浏览器输入这个网址
点开pycharm,新建一个文件,导入visdom
from visdom import Visdom
import numpy as np
vis = Visdom()
vis.text("hello world")
vis.image(np.ones((3,10,10)))