Hrnet调试过程中遇到的坑

    技术2025-08-02  14

    这里写自定义目录标题

    按照官方GitHub搭建运行环境

    按照官方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上训练。

    Processed: 0.028, SQL: 9