from tensorboardX import SummaryWriter
writer = SummaryWriter(args.snapshots_dir)
writer.add_scalar('Train/tr_loss_iter', losses.avg, epoch*len(train_loader)+i)
from tensorboardX import SummaryWriter
writer = SummaryWriter(args.snapshots_dir)
writer.add_scalar('Train/tr_loss_iter', losses.avg, epoch*len(train_loader)+i)