Tensorflow Can';t使用torch.utils.tensorboard查看图表

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

我试着习惯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 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
    
2。改为安装tensorboardX:

  • 卸载张力板:

    tensorboard --logdir runs_pytorch
    
    如果张力板由pip安装:

        pip uninstall tensorboard
    
    如果您的张力板是由anaconda安装的:

        conda uninstall tensorboard
    
  • 安装张紧杆DX

        pip install tensorboardX
    
  • 写剧本的时候,, 改变

    from tensorboardX import SummaryWriter