HRNET使用过程中的问题记录

陈工一

于 2020-04-16 16:16:33 发布

阅读量2.4k

点赞数 3

分类专栏： Python CV 文章标签：深度学习 pytorch 神经网络

本文链接：https://blog.youkuaiyun.com/qq_28123095/article/details/105559727

版权

HRNET使用过程中的问题记录

训练COCO数据集

背景：最近在尝试HRNet，把遇到的一些问题和解决方法简单做个记录~~可能会接着更新

训练COCO数据集

按照官网教程，使用’‘python tools/train.py --cfg experiments/coco/hrnet/w32_256x192_adam_lr1e-3.yaml’'进行训练时，出现错误如下：

Traceback (most recent call last):
  File "tools/train.py", line 223, in <module>
    main()
  File "tools/train.py", line 111, in main
    writer_dict['writer'].add_graph(model, (dump_input, ))
  File "/root/anaconda3/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 "/root/anaconda3/lib/python3.6/site-packages/tensorboardX/pytorch_graph.py", line 240, in graph
    list_of_nodes, node_stats = parse(graph, args, omit_useless_nodes)
  File "/root/anaconda3/lib/python3.6/site-packages/tensorboardX/pytorch_graph.py", line 161, in parse
    nodes_py.append(Node_p