tensorboard打印pytorch网络结构遇到错误

最新推荐文章于 2023-10-19 11:47:32 发布

博博有个大大大的Dream

最新推荐文章于 2023-10-19 11:47:32 发布

阅读量2.9k

点赞数 3

分类专栏：环境配置文章标签： pytorch python 人工智能

本文链接：https://blog.csdn.net/qq_17783559/article/details/113111609

版权

环境配置专栏收录该内容

50 篇文章 13 订阅

订阅专栏

尝试用tensorboard打印pytorch的网络结构。pytorch版本是1.4.0，配置过程中间走了一些弯路记录下来，最终解决方案请参考文末。

1、问题一

初步解决方案：

将

trace, _ = torch.jit.get_trace_graph(model, args)

改为：

trace, _ = torch.jit._get_trace_graph(model, args)

此方案一般。

2、问题二

File "D:\Softwares\Anaconda3\lib\site-packages\tensorboardX\writer.py", line 419, in add_graph
self.file_writer.add_graph(graph(model, input_to_model, verbose))
File "D:\Softwares\Anaconda3\lib\site-packages\tensorboardX\graph.py", line 79, in graph
torch.onnx._optimize_trace(trace, False)
File "D:\Softwares\Anaconda3\lib\site-packages\torch\onnx\__init__.py", line 163, in _optimize_trace
return utils._optimize_graph(graph, operator_export_type)
File "D:\Softwares\Anaconda3\lib\site-packages\torch\onnx\utils.py", line 135, in _optimize_graph
graph = torch._C._jit_pass_onnx(graph, operator_export_type)
TypeError: _jit_pass_onnx(): incompatible function arguments. The following argument types are supported:
1. (arg0: torch::jit::Graph, arg1: torch._C._onnx.OperatorExportTypes) -> torch::jit::Graph

初步解决方案：

将

graph = torch._C._jit_pass_onnx(graph, operator_export_type)

改为：

graph = torch._C._jit_pass_onnx(graph, OperatorExportTypes.ONNX)

此方案一般。

3、问题三

初步解决方案：

升级tensorboardx即可。

pip install --upgrade tensorboardx

此方案一般。

最终解决方案：