安裝
pip install tensorflow
pip install tensorboardx
據(jù)說(shuō)tensorboardx是基于tensorflow的tensorboard的所以要先安裝tensorflow
簡(jiǎn)單使用
from tensorboardX import SummaryWriter
writer = SummaryWriter()
# ...
writer.add_scalar('train_loss', loss, n)
writer.add_graph(model, torch.rand([1,3,224,224]))
可視化結(jié)果命令
tensorboard --logdir=runs
官方文檔https://tensorboardx.readthedocs.io/en/latest/tutorial_zh.html