Skip to main content

Tiu MoLess 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