Визуализация графиков тензорной доски с помощью PyTorch

Question

Я пытаюсь визуализировать график модели PyTorch, используя: https://github.com/lanpa/tensorboardX

Однако, когда я следую инструкциям на странице GitHub README для демонстрационного примера:

• Запустите демонстрационный скрипт: python demo.py

• Используйте TensorBoard с tensorboard --logdir runs (необходимо установить TensorFlow)

Большинство параметров работают - за исключением вкладки График. При переходе туда отображается это сообщение об ошибке:

Файлы определения графа не найдены.

Чтобы сохранить график, создайте tf.summary.FileWriter и передать граф либо через конструктор, либо вызвав его add_graph() метод. Вы можете проверить учебник визуализатора графика.

Если вы новичок в использовании TensorBoard и хотите узнать, как добавить данные и настроить файлы событий, ознакомьтесь с README и, возможно, учебник по TensorBoard.

Если вы считаете, что TensorBoard настроен правильно, см. Раздел README посвященный отсутствующим проблемам с данными, и рассмотрите возможность подачи заявки на GitHub.

Пока оболочка выводит эти ошибки:

W0817 13:51:54.464755 Thread-1 application.py:272] path /data/plugin/debugger/debugger_grpc_host_port not found, sending 404
W0817 13:51:54.493842 Thread-1 application.py:272] path /data/plugin/debugger/debugger_grpc_host_port not found, sending 404
W0817 13:51:54.507973 Thread-1 application.py:272] path /data/plugin/debugger/debugger_grpc_host_port not found, sending 404
W0817 13:51:54.518329 Thread-1 application.py:272] path /data/plugin/debugger/debugger_grpc_host_port not found, sending 404
W0817 13:51:54.527345 Thread-1 application.py:272] path /data/plugin/debugger/comm not found, sending 404
W0817 13:51:54.536046 Thread-1 application.py:272] path /data/plugin/debugger/comm not found, sending 404
W0817 13:51:54.557759 Thread-1 application.py:272] path /data/plugin/debugger/comm not found, sending 404
W0817 13:51:54.570233 Thread-1 application.py:272] path /data/plugin/debugger/comm not found, sending 404

Я установил tensorboadX мастер-ветка с использованием:

pip install git+https://github.com/lanpa/tensorboardX

(Я пробовал также tensorboardX версия 1.2 раньше безуспешно)

Я пытался перейти на другие версии, но безуспешно, в том числе те, которые были "рекомендованы" в сочетании с tensorboardX:

Протестировано на anaconda2 / anaconda3, с PyTorch 0.4 / torchvision 0.2 / тензорная доска 1.7.0

Существует множество похожих проблем с тем же сообщением об ошибке: No graph definition files were found., но все, что я нашел, связано с самим Tensorflow, а не с PyTorch.

Мне кажется, это просто проблема с поиском правильных файлов, но я пока не нашел решения.

Любая помощь очень ценится!