pytorch tensorboard 踩坑记录
实现功能:实时查看loss变化曲线。
代码:
from torch.utils.tensorboard import SummaryWriter
#实例化SummaryWriter对象
writer = SummaryWriter(log_dir=“runs/experiment” )
#写入loss
writer.add_scalar(‘loss’, loss, epoch_num)
tensorboard --logdir=runs
错误:无法发现载入记录
更改:tensorboard --logdir=绝对路径
okay down