pytorch上用tensorboard可视化。
#先安装tensorboard插件
https://github.com/TeamHG-Memex/tensorboard_logger
方法一:
pip install tensorboard_logger
# 测试tensorboard
from tensorboard_logger import configure, log_value configure("runs/run-1234") for step in range(1000): v1, v2 = step,step log_value('v1', v1, step) log_value('v2', v2, step)
方法二: https://github.com/lanpa/tensorboard-pytorch
pip install tensorboardX
from tensorboardX