yolov5s_6.修改记录

文章主要讨论了在Python训练过程中遇到的RuntimeError关于类型转换的问题,以及信号处理模块中不存在SIGALRM属性的AttributeError。提出了通过调整参数如--cache和--batch-size来加速训练,特别是优化内存使用。此外,还涉及了如何处理训练中断,利用--resume参数恢复训练,并设置了早停机制以提高效率。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一.teain.py

——5.0 LOSS报错↓

解决错误:RuntimeError: result type Float can‘t be cast to the desired output type __int64_顾悦西的博客-CSDN博客

——module 'signal' has no attribute 'SIGALRM' ↓

Python错误:AttributeError: module 'signal' has no attribute 'SIGALRM'_polyhedronx的博客-CSDN博客

——训练提速>1/3↓

    # parser.add_argument('--cache', type=str, nargs='?', const='ram', help='--cache images in "ram" (default) or "disk"')#是否提前缓存图片到内存(原)
    parser.add_argument('--cache', action='store_false', help='--cache images in "ram" (default) or "disk"')  ###节省1/3时间(修改)

————目录—文件位置↓

 val: ../data/datasets/images/val  表示在F:\my\yolo\v5中(下)查找val文件

训练集、验证集、测试集的作用和意义_测试集是什么_AI视觉网奇的博客-CSDN博客

——显存↓

'--batch-size', type=int, default=32 (8g显存,32好像不能再调搞了)

parser.add_argument('--batch-size', type=int, default=32, help='total batch size for all GPUs, -1 for autobatch')

——(各种原因)中断训练↓

  开启中断继续训练,应该(没测试)同时需要修改parser.add_argument('--weights', type=str, default=''中断时 【时间最新】的【runs\train\exp4\weights中的】best_yawn.pt或last.pt(*)

    parser.add_argument('--resume', nargs='?', const=True, default=False, help='resume most recent training')  #原(默认)
    # parser.add_argument('--resume', nargs='?', const=True, default=True, help='resume most recent training')  #

——提前自动结束训练: 默认100次不更新就停止训练,早停可改30↓

    parser.add_argument('--patience', type=int, default=30, help='EarlyStopping patience (epochs without improvement)') ##原 default=100

待:

结果可视化——样本

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值