智算集群训练yolov5心得

分享者：crayon

分享地址：

前言：

具体分享一下我在启智平台配置yolov5项目的心得体会，中途也踩了不少坑，也是通过自己不断摸索得出来的经验，希望能对想在该平台运行yolov5的同学有帮助，这里十分推荐大家体验一下智算网络集群，训练速度非常快。

路径设置：

以在智算网络集群运行yolov5为例。首先，得修改train.py文件，将所需要的参数改成自己所需要的，从而不用在参数界面设置也可以正常完成训练任务，可以节省发起训练任务的时间，这里常见的主要包括四个参数：weights、cfg、data、hyp，如图所示。

这里重点要注意的是配置输出目录路径，必须要修改默认project的输出路径，如图所示。

代码注释：

因为下载某些字体时可能会导致训练过程中卡住从而无法正常训练代码，因此必须要把字体检测代码行注释掉，yolov5才可以成功运行，具体需要注释utils/general.py文件里面的一行代码。

关于使用预训练权重：

平台不支持直接上传.pt预训练权重文件，因此经老师们的指导，应该将预训练权重打包在数据集中上传，再修改train.py中的weights路径为数据集路径下的pt文件，小伙伴们可以根据实际情况进行合适的调用。

训练一直显示等待问题：

可能训练人数比较多，耐心等待即可，会自动分配训练任务，不需要重复提交训练任务。

致谢：

最后感谢社区平台提供的免费算力资源，以及老师们的悉心指点，我也会尽我所能为平台出一份力，祝平台发展的越来越好！！