我正在尝试使用以下方法可视化 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 上提交问题。
当 shell 输出这些错误时:
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 / tensorboard 1.7.0
有很多类似的问题,同样的错误消息说明:No graph definition files were found.
,但我发现所有这些问题都与 Tensorflow 本身有关,而不是 PyTorch。
在我看来,这只是定位正确文件的问题,但我还没有找到任何解决方案。
任何帮助是极大的赞赏!