PyTorch with TensorboardX

在构建基于PyTorch的深度神经网络并开展训练工作时，可视化手段对于监控网络训练过程具有关键意义。在实际应用过程中，相较于PyTorch自带的visdom工具，TensorboardX更受到用户的青睐。在参考了网络上众多相关教程的指导下，我最终成功实现了TensorboardX的配置与使用。然而目前网上的教程版本普遍较为陈旧，在实际操作中我遇到了诸多Bug。为便于同学们更好地使用，现将适用于较新版本的操作流程进行整理与总结，希望能为大家提供一定的帮助。本次操作所使用的PyTorch版本为1.1.0，TensorboardX版本为1.6。接下来直接进入教程内容：

1.安装

2.在Pycharm的当前工作目录下创建一个新的logger.py文件

 import tensorflow as tf

    
 import numpy as np
    
 import scipy.misc
    
  
    
 try:
    
     from StringIO import StringIO  # Python 2.7
    
 except ImportError:
    
     from io import BytesIO  # Python