higherHRNet 训练过程出现的问题
在训练higherhnet的时候,按照github上的命令 出现下方箭头的报错,参考网上其他答案,有说按照上面箭头改会解决,但是我改了依旧报相同的错,不知道有没有大佬解决了,求指点。
python tools/dist_train.py --cfg experiments/coco/higher_hrnet/w32_512_adam_lr1e-3.yaml
我好像已经找到这个问题的解决方法了。 具体修改如下,其实只要注释掉train文件里的一行就行,应该是tensorboardx的问题。
/tools/dist_train.py
# writer_dict['writer'].add_graph(model,(dump,input,))