tensorboardX使用中 AttributeError: 'function' object has no attribute 'graph'

最近在使用tensorboardX可视化网络结构，因为tensorboardX并非pytorch原生的可视化工具，所以版本之间并不兼容

在使用的过程中会遇到，AttributeError: 'function' object has no attribute 'graph'，大概是版本之间不兼容

完整错误如下

Traceback (most recent call last):
  File "source_hourglassnet.py", line 181, in <module>
    writer.add_graph(model, (input_data, ))
  File "/home/yongjie/anaconda2/envs/danfeng/lib/python3.6/site-packages/tensorboardX/writer.py", line 738, in add_graph
    self._get_file_writer().add_graph(graph(model, input_to_model, verbose, **kwargs))
  File "/home/yongjie/anaconda2/envs/danfeng/lib/python3.6/site-packages/tensorboardX/pytorch_graph.py", line 241, in graph
    graph = trace.graph
AttributeError: 'function' object has no attribute 'graph'

View Code