pytorch tensorboard

最新推荐文章于 2024-07-19 16:12:05 发布

weixin_42673583

最新推荐文章于 2024-07-19 16:12:05 发布

阅读量287

点赞数 1

分类专栏：机器学习文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/weixin_42673583/article/details/126301271

版权

机器学习专栏收录该内容

6 篇文章 0 订阅

订阅专栏

import random

from torch.utils.tensorboard import SummaryWriter#导入类
writer=SummaryWriter(“d://log”)#类实例化，传入存储事件文件夹的名称
#writer.add_image()
#y=3x
epoch = 500
for i in range(epoch):

writer.add_scalars('ssim-model1', {'ssim-15%': 0.8+random.uniform(0.01,0.02)}, i)
writer.add_scalars('ssim-model1', {'ssim-20%': 0.7+random.uniform(0.01,0.02)}, i)
writer.add_scalars('ssim-model1', {'ssim-50%': 0.6+random.uniform(0.01,0.02)}, i)

for i in range(epoch):
writer.add_scalars(‘loss-model1’, {‘loss-15%’: 0.231+random.uniform(0.01,0.02)}, i)
writer.add_scalars(‘loss-model1’, {‘loss-20%’: 0.531+random.uniform(0.01,0.02)}, i)
writer.add_scalars(‘loss-model1’, {‘loss-50%’: 0.321+random.uniform(0.01,0.02)}, i)

for i in range(epoch):
writer.add_scalars(‘ssim-model2’, {‘ssim-15%’: 0.82+random.uniform(0.01,0.02)}, i)
writer.add_scalars(‘ssim-model2’, {‘ssim-20%’: 0.88+random.uniform(0.01,0.02)}, i)
writer.add_scalars(‘ssim-model2’, {‘ssim-50%’: 0.7+random.uniform(0.01,0.02)}, i)

for i in range(100):
writer.add_scalars(‘loss-model2’, {‘loss-15%’: 0.271+random.uniform(0.01,0.02)}, i)
writer.add_scalars(‘loss-model2’, {‘loss-20%’: 0.421+random.uniform(0.01,0.02)}, i)
writer.add_scalars(‘loss-model2’, {‘loss-50%’: 0.211+random.uniform(0.01,0.02)}, i)

writer.close()