pytorch 利用tensorboard显示loss，acc曲线等

版本信息：
python 3.6
pytorch 0.4.1
tensorflow-gpu 1.4.0
tensorboard 1.6.0 [tensorflow 与 tensorboard版本匹配]

from tensorboard_logger import Logger

logger = Logger(logdir="./tensorboard_logs", flush_secs=10)
...
def train(net, optimizer):
    for epoch in range(epoch_nums):
        net.train()
        for batch_idx, (inputs, targets) in enumerate(trainloader):           
            inputs = Variable(inputs, requires_grad=True).cuda()
            targets = targets.cuda()
            optimizer.zero_grad()
            outputs = net(inputs)
            loss = criterion(outputs, targets)
            loss.backward()  
            optimizer.step()
            train_loss += loss.item()
            ...
            # 记录所需的变量
            logger.log_value('avg_loss', train_loss/(batch_idx+1), epoch*len(trainloader) + batch_idx)
            logger.log_value('loss', loss.item(), epoch*len(trainloader) + batch_idx)
            logger.log_value('acc', 100. * correct / total, epoch*len(trainloader) + batch_idx)

新开一个终端，输入：

tensorboard --logdir './tensorboard_logs/'

回车后终端显示：

TensorBoard 1.6.0 at http://iccd:6006 (Press CTRL+C to quit)

地址粘到浏览器中：可以看到所需的曲线了 😃