pytorch调用tensorboard方法尝试

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/yutingzhaomeng/article/details/82380701

tensorboard提供了用于监视训练损失很好的接口，可以帮助我们更好的调整参数。下文介绍如何在pytorch中调用tensorboard。

首先，安装tensorboard、tensorflow以及tensorboardX

第二，在文件开头导入SummaryWriter

from tensorboardX import SummaryWriter

第三，同tensorflow的tensorboard一样，tensorboardX提供多种记录方式如scalar、image等。

writer = SummaryWriter('path')

如果不添加path，则默认以时间命名。

第四，添加监视变量

writer.add_scalar('Train/Acc', Acc, iter)

第五，打开tensorboard

tensorboard --logdir 'path'

第六，在浏览器打开6006端口