Я пытаюсь привыкнуть к тензорной доске и кодирую свои модели, используя 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 --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
Большое спасибо за вашу помощь, это сводит меня с ума!