1.Tensorboard的显示
选择需要可视化的数据,如损失、奖励等。。。。。。
(1)安装tensorboard包,自己安就行,不会就问你师兄
(2)你的程序里写上如下代码
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter(log_dir='runs/mock_accuracy')#目录和名字自己设置
(3)找个没人的地,打开对应的路径,cmd,conda activate “你的虚拟环境”
(4)tensorboard --logdir runs(刚才的那个路径)
(5)点开http:…
(6)图中的浅色线和深色线是因为smoothing不为0,调一下就行2.一个tag多个图
writer.add_scalar(tag=‘TrainLoss/a’, scalar_value=i * random.uniform(1, 5), global_step=i)
writer.add_scalar(tag=‘TrainLoss/c’, scalar_value=i * random.uniform(1, 5), global_step=i)
writer.add_scalar(tag=‘TrainLoss/b’, scalar_value=i * random.uniform(1, 5), global_step=i)