栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

YOLOX接着以前训练的模型再训练

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

YOLOX接着以前训练的模型再训练

觉得当前训练的epoch数量还不够,想要再多训练几个epoch,又不想从头开始训练,那下面的方法很适合你:
  • 1. 修改train.py
    parser.add_argument(
    "--resume", default=True, action="store_true", help="resume training"					#改为True
    )
    parser.add_argument("-c", "--ckpt",
    default='YOLOX_outputs/yolox_voc_s/latest_ckpt.pth', type=str, help="checkpoint file")	#预训练权重用最后一个epoch的权重
    parser.add_argument(
    "-e",
    "--start_epoch",
    default=100,			#从第100epoch开始,根据自己的实际情况修改
    type=int,
    help="resume training start epoch",
    )
    
  • 2. 也可直接在命令行操作: python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 4 -c YOLOX_outputs/yolox_voc_s/latest_ckpt.pth.tar -resume -start_epoch=100
  • 再次训练,就可以发现你的epoch不是从1开始了
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1017774.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号