【mmdetection中的--cfg-options使用教程】

一文了解mmdetection中的–cfg-options

--cfg-options是用于在命令行中设置mmdetection配置文件的选项。它允许您以键值对的形式覆盖配置文件中的选项。

以下是使用--cfg-options的示例:

python tools/train.py ${CONFIG_FILE} --cfg-options \
    optimizer.lr=0.02 \
    optimizer.weight_decay=0.0001 \
    runtime.total_epochs=12

此命令将使用给定的${CONFIG_FILE}作为基础配置,并覆盖三个选项:optimizer.lroptimizer.weight_decayruntime.total_epochs

其中,optimizer.lroptimizer.weight_decay是优化器的超参数,runtime.total_epochs是训练总轮数。
您可以将任意数量的选项传递给--cfg-options,每个选项之间使用空格分隔。如果选项值包含空格,则应使用双引号将其括起来。
请注意,这些设置将优先于配置文件中的设置。因此,在覆盖选项时,最好了解配置文件中定义的默认值。

优化器修改

配置文件:

optim_wrapper = dict(type=‘OptimWrapper’, 
					optimizer=dict(type=‘Adam’, 
					lr=0.0003, 
					weight_decay=0.0001))

将优化器使用命令行修改配置文件

python tools/train.py configs/my_config.py 
		--cfg-options optim_wrapper.optimizer.type=‘Adam’ 
		optim_wrapper.optimizer.lr=0.0003

学习率策略修改

配置文件:

param_scheduler = [
    dict(type='LinearLR', 
    	start_factor=0.001, 
    	by_epoch=False, 
    	begin=0, 
    	end=500),
    dict(type='MultiStepLR',
        begin=0,
        end=24,
        by_epoch=True,
        milestones=[16, 22],
        gamma=0.1)
]

将学习率策略使用命令行修改配置文件

python tools/train.py configs/my_config.py 
			–cfg-options param_scheduler.0.start_factor=0.01 
			param_scheduler.1.milestones="[12, 18]"

设置训练轮数

配置文件:

train_cfg = dict(type='EpochBasedTrainLoop', max_epochs=24, val_interval=1)

将训练轮数使用命令行修改配置文件

python tools/train.py configs/my_config.py 
			–cfg-options train_cfg.max_epochs=100
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Zhijun.li@Studio

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值