当前位置:网站首页>MMDetection finetune
MMDetection finetune
2022-08-04 07:01:00 【子燕若水】
https://github.com/open-mmlab/mmdetection/blob/master/docs/zh_cn/tutorials/finetune.md
里面的一些参数含义:
optimizer = dict( # 构造optimizer的配置, 支持PyTorch中所有的优化器,并且参数名称也和PyTorch中提供的一样。
type='SGD', # optimizers的类型, 具体细节请参照https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/optimizer/default_constructor.py#L13.
lr=0.02, # optimizers的学习率, 请到PyTorch的文档中查看相关参数的具体用法。
momentum=0.9, # SGD优化器的超参数:Momentum
weight_decay=0.0001) # SGD优化器的超参数:Weight decay
optimizer_config = dict( # 构造optimizer hook的配置, 具体细节请参照 https://github.com/open-mmlab/mmcv/blob/master/mmcv/runner/hooks/optimizer.py#L8.
grad_clip=None) # 绝大多少方法都不会使用gradient clip
lr_config = dict( # Learning rate scheduler config used to register LrUpdater hook
policy='step', # The policy of scheduler, also support CosineAnnealing, Cyclic, etc. Refer to details of supported LrUpdater from https://github.com/open-mmlab/mmcv/blob/master/mmcv/runner/hooks/lr_updater.py#L9.
warmup='linear', # warmup的策略, 还支持 `exp` 和 `constant`.
warmup_iters=500, # warmup的迭代次数
warmup_ratio=
0.001, # 用于warmup的起始学习比率
step=[8, 11]) # 学习率进行衰减的step位置
total_epochs = 12 # model训练的总epoch数
checkpoint_config = dict( # 设置checkpoint hook, 具体细节请参照https://github.com/open-mmlab/mmcv/blob/master/mmcv/runner/hooks/checkpoint.py 的实现.
interval=1) # 每隔几个epoch保存一下checkpoint
log_config = dict( # logger文件的配置
interval=50, # 每隔多少个epoch输出一个log文件
hooks=[
# dict(type='TensorboardLoggerHook') # MMDetection支持Tensorboard logger
dict(type='TextLoggerHook')
]) # logger 被用来记录训练过程.
dist_params = dict(backend='nccl') # 设置分布式训练的参数,也可以设置端口。
log_level = 'INFO' # The level of logging.
load_from = None # 给出之前预训练模型checkpoint的路径,这个不会resume training(resume training会按照上次的记录接着训练,而这个参数应该只是导入之前预训练模型参数,重新训练)
resume_from = None # 给出需要Resume 的checkpoints的路径, 它将会接着上次被保存的地方进行训练。
workflow = [('train', 1)] # Workflow for runner. [('train', 1)] means there is only one workflow and the workflow named 'train' is executed once. The workflow trains the model by 12 epochs according to the total_epochs.(这个workflow具体是干什么的我不是很清楚orz)
work_dir = 'work_dir' # 保存模型的文件夹路径(checkpoints和log文件都会保存在其中)。
边栏推荐
猜你喜欢
[Paper Notes] - Low Illumination Image Enhancement - Supervised - RetinexNet - 2018-BMVC
Error ER_NOT_SUPPORTED_AUTH_MODE Client does not support authentication protocol requested by serv
一天学会JDBC04:ResultSet的用法
RT-Thread Studio学习(十二)W25Q128(SPI)的读写
千万级别的表分页查询非常慢,怎么办?
小程序如何使用订阅消息(PHP代码+小程序js代码)
MySQL内存淘汰策略
七夕送礼,心愿直抵!
字节跳动岗位薪酬体系曝光,看完我真的酸了...
fanuc机器人IO分配报警信号分配无效
随机推荐
SQL存储过程详解
8.2学习记录
两日总结八
CSDN21天学习挑战赛——day1 正则表达式大总结
Amazon亚马逊 Vendor Central Label详解
MySQL内存淘汰策略
data:image/jpg;base64格式数据转化为图片
Praat:语音标注工具【保存为TextGrid文件】
LeetCode 135. 分发糖果
SQL去重的三种方法汇总
电商系统PC商城模块介绍
专属程序员的浪漫七夕
ExoPlayer添加Ffmpeg扩展实现软解功能
【字符串】最小表示法
2022年7月总结
MySQL重置root密码
Produce definition 产品与行业分析 勤于思考 善于总结 强于表达
解决:Hbuilder工具点击发行打包,一直报尚未完成社区身份验证,请点击链接xxxxx,项目xxx发布H5失败的错误。
ContrstrainLayout的动画之ConstraintSet
MySQL BIGINT 数据类型