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

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

运行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

参考一下