@pengsun there are some scripts provided in this thread: https://www.reddit.com/r/MachineLearning/comments/5pbdnj/d_visualizing_training_with_pytorch/ 805
Particularly, look at:
https://github.com/TeamHG-Memex/tensorboard_logger 471