tensorboard-可视化可视化pytorch网络模型网络模型
1.Anaconda环境配置
①pytorch 1.4
②tensorflow(2.0版本及对应的tensorboard)
③tensorboardX(采用pytorch自带的,可能不需要再安装)
2.SummaryWriter写出需要的log信息
①采用torch.utils.tensorboard下的SummaryWriter
②示例代码如下:
③程序运行后在程序所在文件夹上生成runs文件夹
3.执行tensorboard命令
采用Anaconda Prompt执行
采用pycharm执行
评论1