运行HigherHRNet官方源码的时候，训练dis_train.py程序报错，bug怎么改正？

Question

问题遇到的现象和发生背景

运行HigherHRNet官方源码的时候，训练dis_train.py程序报错

问题相关代码，请勿粘贴截图

HigherHRNet开源代码

运行结果及报错内容

Traceback (most recent call last):
  File "tools/dist_train.py", line 326, in <module>
    main()
  File "tools/dist_train.py", line 117, in main
    args=(ngpus_per_node, args, final_output_dir, tb_log_dir)
  File "/home/huyuhang/anaconda3/envs/HRNet/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 171, in spawn
    while not spawn_context.join():
  File "/home/huyuhang/anaconda3/envs/HRNet/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 118, in join
    raise Exception(msg)
Exception: 

-- Process 0 terminated with the following error:
Traceback (most recent call last):
  File "/home/huyuhang/anaconda3/envs/HRNet/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 19, in _wrap
    fn(i, *args)
  File "/home/huyuhang/HigherHRNet-Human-Pose-Estimation-master/tools/dist_train.py", line 203, in main_worker
    writer_dict['writer'].add_graph(model, (dump_input, ))
  File "/home/huyuhang/anaconda3/envs/HRNet/lib/python3.6/site-packages/tensorboardX/writer.py", line 566, in add_graph
    self.file_writer.add_graph(graph(model, input_to_model, verbose))
  File "/home/huyuhang/anaconda3/envs/HRNet/lib/python3.6/site-packages/tensorboardX/pytorch_graph.py", line 235, in graph
    _optimize_trace(trace, torch.onnx.utils.OperatorExportTypes.ONNX)
  File "/home/huyuhang/anaconda3/envs/HRNet/lib/python3.6/site-packages/tensorboardX/pytorch_graph.py", line 175, in _optimize_trace
    trace.set_graph(_optimize_graph(trace.graph(), operator_export_type))
AttributeError: 'torch._C.Graph' object has no attribute 'set_graph'

我的解答思路和尝试过的方法

我想要达到的结果

改bug

Answer 1

参考一下

tensorboard打印pytorch网络结构遇到错误_博博有个大大大的Dream的博客-CSDN博客 尝试用tensorboard打印pytorch的网络结构。pytorch版本是1.4.0，配置过程中间走了一些弯路记录下来，最终解决方案请参考文末。1、问题一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 "... https://blog.csdn.net/qq_17783559/article/details/113111609