Не могу увидеть график, используя torch.utils.tensorboard - PullRequest
1 голос
/ 10 октября 2019

Я пытаюсь привыкнуть к тензорной доске и кодирую свои модели, используя pytorch. Однако, когда я пытаюсь увидеть мою модель с помощью функции add_graph (), я получаю это: enter image description here

С этим в качестве тестового кода:

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, все выглядит нормально: enter image description here

с этим в качестве тестового кода на этот раз:

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 --logdir runs_pytorch

Что-то, что я заметил, это то, что когда я использую его в каталоге, выделенном для моего теста тензорного потока, яполучил обычное сообщение с адресом, но если я сделаю то же самое с --logdir run_pytorch, у меня получится нечто большее:

W1010 15: 19: 24.225109 15308 plugin_event_accumulator.py:294] Найдено более одногографическое событие за прогон, или существовал метаграф, содержащий graph_def, а также одно или несколько графических событий. Перезапись графа новейшим событием. W1010 15: 19: 24.226075 15308 plugin_event_accumulator.py:322] Найдено более одного события «Выполнить метаданные» с тегом step1. Перезаписываем его новым событием.

Я на Windows, я пробовал в разных браузерах (Chrome, Firefox ...). У меня есть tenorflow 1.14.0, torch 1.2.0, Python 3.7.3

Большое спасибо за вашу помощь, это сводит меня с ума!

Ответы [ 2 ]

1 голос
/ 21 октября 2019

Есть два способа ее решения:

1. Обновите PyTorch до версии 1.3.0 и выше:

2. вместо этого установите тензорную доскуX:

  • удалите тензорную доску:

    если ваша тензорная доска установлена ​​в пипсах:

        pip uninstall tensorboard
    

    если вашатензорная доска установлена ​​anaconda:

        conda uninstall tensorboard
    
  • установить тензорную доскуX

        pip install tensorboardX
    
  • при написании сценария измените

        from torch.utils.tensorboard import SummaryWriter
    

    до

        from tensorboardX import SummaryWriter
    
0 голосов
/ 11 октября 2019

Это могло быть вызвано этой известной проблемой , и кажется, что она была решена в pytorch 1.3, который был выпущен вчера - посмотрите Исправления ошибок в выпуске примечания .

...