admin管理员组文章数量:1656657
运行之前确保设定正确如下几个参数。
parser.add_argument("--weights", type=str, default="weights/yolov5l.pt" , help="initial weights path")
parser.add_argument("--cfg", type=str, default="models/yolov5l.yaml", help="model.yaml path")
parser.add_argument("--data", type=str, default="data/coco128.yaml" , help="dataset.yaml path")
parser.add_argument("--batch-size", type=int, default=4, help="total batch size for all GPUs, -1 for autobatch")
parser.add_argument("--workers", type=int, default=4, help="max dataloader workers (per RANK in DDP mode)")
建议刚开始时batch-size和workers的参数设置小一点,数值过大会导致卡顿或者运行缓慢。
笔者使用的是4G的独显,参数均设置为4-6,可正常运行。
正常启动后,如果运行比较慢或者卡顿。先保持程序运行,并且进入任务管理器检查设备运行情况。
如果设置的是用CPU或者特定GPU进行训练,注意观察负载,如果不符合预期,则需要检查以下代码。
parser.add_argument("--device", default=0, help="cuda device, i.e. 0 or 0,1,2,3 or cpu")
对应参数设置为任务管理器中硬件编号后方可正常运行。比如期望使用GPU0,则设置为default=0。
版权声明:本文标题:使用YOLOV5进行训练时电脑崩溃或运行缓慢如何解决 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://m.elefans.com/dianzi/1729745861a1211719.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论