美文网首页
2020-03-30 pytorch中的tensorboardx

2020-03-30 pytorch中的tensorboardx

作者: Hie_9e55 | 来源:发表于2020-03-30 10:22 被阅读0次

安装

pip install tensorflow
pip install tensorboardx
据说tensorboardx是基于tensorflow的tensorboard的所以要先安装tensorflow

简单使用

from tensorboardX import SummaryWriter   
writer = SummaryWriter()
# ...
writer.add_scalar('train_loss', loss, n)
writer.add_graph(model, torch.rand([1,3,224,224]))

可视化结果命令

tensorboard --logdir=runs
官方文档https://tensorboardx.readthedocs.io/en/latest/tutorial_zh.html

相关文章

网友评论

      本文标题:2020-03-30 pytorch中的tensorboardx

      本文链接:https://www.haomeiwen.com/subject/rcumuhtx.html