深度学习与PyTorch笔记21

最新推荐文章于 2022-11-15 19:40:24 发布

niuniu990

最新推荐文章于 2022-11-15 19:40:24 发布

阅读量492

点赞数

文章标签： python pytorch

本文链接：https://blog.csdn.net/niuniu990/article/details/88560430

版权

Visdom可视化

TensorboardX

使用pip install tensorboardX命令来安装。
tensorboardX在pytorch中怎么使用：
它有一个SummaryWriter，新建一个SummaryWriter实例，把要监听的数据起一个名字，(‘data/scalar’（名字）,dummy_s1[0]（数据）,n_iter（x坐标，代表的是哪一个epoch）)。
tensorboard本质上抽取的是一个numpy的数据，如果要和tensor做一个match的话，必须先把tensor转换到cpu上面（.cpu()），然后再转化成numpy数据（data.numpy()）,才能够赋值给tensorboard。

from tensorboardX import SummaryWriter
writer=SummaryWriter()
writer.add_scalar('data/scalar',dummy_s1[0],n_iter)
writer.add_scalar('data/scalar_group',{'xsinx':n_iter*np.sin(n_iter),
                                       'xcosx':n_iter*np.cos(n_iter),
                                       'arctanx':np.arctan(n_iter),},n_iter)
writer.add_image('Image',x,n_iter)
writer.add_text('Text','text logged at step:'+str(n_iter),n_iter)

for name,param in resnet18.named_parampython -m visdom.servereters():
    writer.add_histogram(name,param.clone().cpu().data.numpy(),n_iter)
writer.close()

Visdom from Facebook

使用pip install visdom命令来安装。安装完成后开启监听的进程，确保程序运行前开启visdom，使用python -m visdom.server来开启。
windows开启时容易出现错误：
在这里插入图片描述
解决这个错误的方法就是重新安装visdom。首先pip uninstall visdom卸载之前的visdom，然后从官方网页上下载源代码，解压，然后进入到那个目录下面去cd 目录名&