您的位置:首页 > 其它

Hrnet调试过程中遇到的坑

2020-07-14 06:35 876 查看

这里写自定义目录标题

按照官方GitHub搭建运行环境

关于环境的搭建请仔细阅读readme,首先,我的机器是ubantu16.4,cuda10.0.130。根据pytorch的官网下载,我第一次就下载的torch1.2.0,torchvision0.4.0,tensorboardX1.6,但是训练时遇到了如下问题:

在这个坑里来回试了很久,在pytorch的pytorchIssues中找到了解决办法:
把Torch和torchvision卸载,装上如下版本:
torch1.0.0,torchvision0.2.0,tensorboardX1.6
再用所给的终端命令python tools/train.py
–cfg experiments/coco/hrnet/w32_256x192_adam_lr1e-3.yaml \ 输入即可在coco上训练。

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: