敲代码的小风

from torch.utils.tensorboard import SummaryWriterfrom torch.utils.tensorboard import SummaryWriterimport torchimport torchvisionfrom torchvision import datasets,transformsfrom torch.autograd import Variablefrom ssd import SSDssd = SSD()model = ssd

代码展示:在这里插入代码片控制台输出结果:在这里插入代码片

参考链接: add_hparams代码:from torch.utils.tensorboard import SummaryWriterimport timewith SummaryWriter() as w: for i in range(5): w.add_hparams({'lr': 0.1*i, 'bsize': i}, {'超参数hparam/准确率accuracy': 10*i,

参考链接: add_mesh代码:from torch.utils.tensorboard import SummaryWriterimport torchvertices_tensor = torch.as_tensor([ [1, 1, 1], [-1, -1, 1], [1, -1, -1], [-1, 1, -1],], dtype=torch.float).unsqueeze(0)colors_tensor = torch.as_tensor([

参考链接: add_pr_curve代码:from torch.utils.tensorboard import SummaryWriterimport numpy as npnp.random.seed(20200910)labels = np.random.randint(2, size=100) # binary labelpredictions = np.random.rand(100)writer = SummaryWriter()writer.add_pr_curve('pr_

参考链接: add_embedding参考链接: pytorch 中使用tensorboard，详解writer.add_embedding函数的作用（一）代码展示:from torch.utils.tensorboard import SummaryWriterimport keywordimport torchwriter = SummaryWriter()meta = []while len(meta)<100: meta = meta+keyword.kwlist #

参考链接: add_graph代码:from torch.utils.tensorboard import SummaryWriterimport torchimport torchvisionfrom torchvision import datasets,transformsfrom torch.autograd import Variableclass Model(torch.nn.Module): def __init__(self): super(Mode

参考链接: add_text代码:from torch.utils.tensorboard import SummaryWriterwriter = SummaryWriter()for i in range(3): writer.add_text('lstm', 'This is an lstm', i) writer.add_text('rnn', 'This is an rnn', 10+i)writer.close()# tensorboard --logdir=

参考链接: add_figure代码:from torch.utils.tensorboard import SummaryWriter# 绘制散点图import numpy as npimport matplotlib.pyplot as pltnp.random.seed(20200910)fig, ax = plt.subplots() ax.plot(10*np.random.randn(100),10*np.random.randn(100),'o')ax.set_title

参考链接: add_image参考链接: add_images代码:from torch.utils.tensorboard import SummaryWriterimport numpy as npimg = np.zeros((3, 100, 100))img[0] = np.arange(0, 10000).reshape(100, 100) / 10000img[1] = 1 - np.arange(0, 10000).reshape(100, 100) / 10000img_H

参考链接: add_histogram代码:from torch.utils.tensorboard import SummaryWriterimport numpy as npwriter = SummaryWriter()for i in range(5): # 8 x = np.random.random(1000000) # 1000 writer.add_histogram('林麻子distribution centers', x + 2 * i, i)writer.

参考链接: add_scalar代码:from torch.utils.tensorboard import SummaryWriterwriter = SummaryWriter()x = range(100)for i in x: writer.add_scalar('林祖泉y=2x林麻子', i * 2, i)writer.close()# tensorboard --logdir=runs截图展示:

参考链接: torch.utils.tensorboard.writer.SummaryWriter使用方式torch.utils.tensorboard.writer.SummaryWriter( log_dir=None, comment='', purge_step=None, max_queue=10, flush_secs=120, filename_suffix='')from torch.utils.tensorboard import SummaryWriter

代码:from torch.utils.tensorboard import SummaryWriterimport numpy as npnp.random.seed(20200910)writer = SummaryWriter()for n_iter in range(100): writer.add_scalar('Loss/train', np.random.random(), n_iter) writer.add_scalar('Loss/test', np.rand

参考链接: torch.utils.tensorboard安装:pip install tensorboard或者安装tensorflow,默认CPU版本即可:pip install tensorflow运行:tensorboard --logdir=runs