tensorboardX使用

from tensorboardX import SummaryWriter

writer = SummaryWriter(args.snapshots_dir)

writer.add_scalar('Train/tr_loss_iter', losses.avg, epoch*len(train_loader)+i)

 

你可能感兴趣的:(tensorboardX)