pytorch tensorboard

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)
1
2
3

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()