Tensorflow Can';t使用torch.utils.tensorboard查看图表
我试着习惯tensorboard,我用pytorch为我的模型编码。 但是,当我尝试使用add_graph()函数查看我的模型时,我得到了以下结果: 以此作为测试代码:Tensorflow Can';t使用torch.utils.tensorboard查看图表,tensorflow,graph,pytorch,tensorboard,Tensorflow,Graph,Pytorch,Tensorboard,我试着习惯tensorboard,我用pytorch为我的模型编码。 但是,当我尝试使用add_graph()函数查看我的模型时,我得到了以下结果: 以此作为测试代码: import numpy as np import torch import torchvision.transforms as transforms import torch.nn as nn import torch.optim as optim from torch.utils.tensorboard import Sum
import numpy as np
import torch
import torchvision.transforms as transforms
import torch.nn as nn
import torch.optim as optim
from torch.utils.tensorboard import SummaryWriter
class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.linear = nn.Linear(2, 1)
def forward(self, x):
x = self.linear(x)
return x
writer = SummaryWriter('runs_pytorch/test')
net = Net()
criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)
writer.add_graph(net, torch.zeros([4, 2], dtype=torch.float))
writer.close()
另一方面,如果我尝试使用TensorFlow查看图形,一切似乎都很好:
这次将此作为测试代码:
import tensorflow as tf
tf.Variable(42, name='foo')
w = tf.summary.FileWriter('runs_tensorflow/test')
w.add_graph(tf.get_default_graph())
w.flush()
w.close()
如果您想知道,我正在使用以下命令启动tensorboard:
tensorboard --logdir runs_pytorch
我注意到,当我在为tensorflow测试分配的目录上使用它时,我得到了带有地址的常见消息,但是如果我对--logdir runs_pytorch做同样的事情,我得到了更多:
W1010 15:19:24.225109 15308 plugin\u event_u累加器.py:294]每次运行时发现多个图形事件,或者有一个包含图形定义的元图以及一个或多个图形事件。用最新事件覆盖图形。
W1010 15:19:24.226075 15308 plugin_event_acculator.py:322]找到多个带有标记step1的“运行元数据”事件。用最新事件覆盖它
我在windows上,我尝试了不同的浏览器(chrome、firefox…)。
我有tensorflow 1.14.0、torch 1.2.0和Python 3.7.3
非常感谢你的帮助,我快疯了 这可能是由此引起的,似乎在昨天重新发布的pytorch 1.3中解决了此问题-请查看中的错误修复。有两种解决方法: 1。将PyTorch更新至1.3.0及以上版本:
- 康达方式:
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch
- pip方式:
pip3 install torch==1.3.0+cu92 torchvision==0.4.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html
- 卸载张力板:
如果张力板由pip安装:tensorboard --logdir runs_pytorch
如果您的张力板是由anaconda安装的:pip uninstall tensorboard
conda uninstall tensorboard
- 安装张紧杆DX
pip install tensorboardX
- 写剧本的时候,, 改变 到
from tensorboardX import SummaryWriter