tensorboard 可视化pytorch网络模型
1.Anaconda环境配置 1pytorch 1.4 2tensorflow(2.0版本及对应的tensorboard) 3tensorboardX(采用pytorch自带的,可能不需要再安装) 2.SummaryWriter写出需要的log信息 1采用torch.utils.tensorboard下的SummaryWriter 2示例代码如下: 3程序运行后在程序所在文件夹上生成runs文件夹 3.执行tensorboard命令 采用Anaconda Prompt执行 采用pycharm执行 4.查看,在浏览器中输入对应网址:http://localhost:6006/ 5
用户评论