Tensorboard

#1)导入 tensorboard,实例化 SummaryWriter 类,指明记录日记路径等信息
#from torch.utils.tensorboard import SummaryWriter
# 实例化 SummaryWriter,并指明日志存放路径。在当前目录如果没有 logs 目录将自动创建
#writer = SummaryWriter(log_dir = 'logs')
# 调用实例
#writer.add_xxx()
# 关闭 writer
#writer.close()

#2)调用相应的 API,接口一般格式为:
#add_xxx(tag_name, object, iteration-number)

#3) 启动 tensorboard,在命令行中输入
#tensorboard --logdir=r’加 logs 所在路径’

#4) 复制网址在浏览器中打开

单条曲线

'''
#单条曲线(scalar)  add_scalar(tag, scalar_value, global_step=None, walltime=None)

tag (string) – 数据标识符
scalar_value (float 或 string/blobname) – 要保存的值
global_step (int) – 要记录的全局步长值
walltime (float) – 记录训练的时间,默认 walltime (time.time ()) 秒
new_style (boolean) – 是使用新样式(张量字段)还是旧样式(simple_value 字段)。新样式可能会导致更快的数据加载。
'''

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('runs')

for x in range(1,11):
    writer.add_scalar('y = 2x',3*x,x)
writer.close()

可视化最简单的办法是租服务器Autodl

Tensorboard_第1张图片

 Tensorboard_第2张图片

 

直方图

from torch.utils.tensorboard import SummaryWriter
import numpy as np
writer = SummaryWriter()
for step in range(10) :
    x = np.random.randn(1000)
    writer.add_histogram('distribution of gaussion', x, step)

writer.close()

Tensorboard_第3张图片

 

打印网络

#1)导入 tensorboard,实例化 SummaryWriter 类,指明记录日记路径等信息
#from torch.utils.tensorboard import SummaryWriter
# 实例化 SummaryWriter,并指明日志存放路径。在当前目录如果没有 logs 目录将自动创建
#writer = SummaryWriter(log_dir = 'logs')
# 调用实例
#writer.add_xxx()
# 关闭 writer
#writer.close()

#2)调用相应的 API,接口一般格式为:
#add_xxx(tag_name, object, iteration-number)

#3) 启动 tensorboard,在命令行中输入
#tensorboard --logdir=r’加 logs 所在路径’

#4) 复制网址在浏览器中打开

'''
#单条曲线(scalar)  add_scalar(tag, scalar_value, global_step=None, walltime=None)

tag (string) – 数据标识符
scalar_value (float 或 string/blobname) – 要保存的值
global_step (int) – 要记录的全局步长值
walltime (float) – 记录训练的时间,默认 walltime (time.time ()) 秒
new_style (boolean) – 是使用新样式(张量字段)还是旧样式(simple_value 字段)。新样式可能会导致更快的数据加载。
'''

from torch.utils.tensorboard import SummaryWriter
import numpy as np
import torch
import torch.nn as nn

writer = SummaryWriter()
class MLP(nn.Module) :
    def __init__(self):
        super(MLP, self).__init__()
        self.Net = nn.Sequential(
            nn.Linear(784, 512),
            nn.ReLU(),

            nn.Linear(512, 128),
            nn.ReLU(),

            nn.Linear(128, 10)
        )
    def forward(self, input):
        input = input.view(-1, 28 * 28)
        return self.Net(input)
model = MLP()
input = torch.FloatTensor(np.random.rand(32, 1, 28, 28))
writer.add_graph(model, input)

writer.close()

Tensorboard_第4张图片

 详解Tensorboard及使用教程_进击的南方仔的博客-CSDN博客_tensorboard目录一、什么是Tensorboard二、配置Tensorboard环境要求安装三、Tensorboard的使用使用各种add方法记录数据单条曲线(scalar)多条曲线(scalars)直方图(histogram)图片(image)渲染(figure)网络(graph)其他三、结语一、什么是TensorboardTensorboard原本是Google TensorFlow的可视化工具,可以用于记录训练数据、评估数据、网络结构、图像等,并且可以在web上展示,对于观察神经网络的过程非常有帮助。PyTorhttps://blog.csdn.net/weixin_53598445/article/details/121301078

你可能感兴趣的:(图像分类,pytorch,人工智能,python)