Less than 1 minute
1、安装tensorboardX: pip install tensorboardX
2、添加实例:
from tensorboardX import SummaryWriter
writer = SummaryWriter('./log')
3、启动端口:tensorboard --logdir=./log 此处可能会报错,大概率是因为没有安装tensorflow,因为tensorboard是依靠TF运行的 使用 pip install tensorflow 安装即可。
4、常用可视化,折线图:
add_scalar(tag, salar_value, global_step=None, walltime=None)
实例:
for batch_idx, (imgs, pids, _) in enumerate(trainloader):
dosomething~~
if (batch_idx + 1) % args.print_freq == 0:
print('Batch {}/{}\t Loss vel:{:.6f} avg:{:.6f}'.format(batch_idx + 1,
len(trainloader),
losses.val,
losses.avg))
writer.add_scalar('loss', losses.avg, epoch + tensorboard_idx)
tensorboard_idx += 0.2