mirror of https://github.com/open-mmlab/mmyolo.git
30 lines
724 B
Python
30 lines
724 B
Python
_base_ = './yolov6_s_syncbn_fast_8xb32-400e_coco.py'
|
|
|
|
max_epochs = 300
|
|
num_last_epochs = 15
|
|
|
|
default_hooks = dict(
|
|
param_scheduler=dict(
|
|
type='YOLOv5ParamSchedulerHook',
|
|
scheduler_type='cosine',
|
|
lr_factor=0.01,
|
|
max_epochs=max_epochs))
|
|
|
|
custom_hooks = [
|
|
dict(
|
|
type='EMAHook',
|
|
ema_type='ExpMomentumEMA',
|
|
momentum=0.0001,
|
|
update_buffers=True,
|
|
strict_load=False,
|
|
priority=49),
|
|
dict(
|
|
type='mmdet.PipelineSwitchHook',
|
|
switch_epoch=max_epochs - num_last_epochs,
|
|
switch_pipeline=_base_.train_pipeline_stage2)
|
|
]
|
|
|
|
train_cfg = dict(
|
|
max_epochs=max_epochs,
|
|
dynamic_intervals=[(max_epochs - num_last_epochs, 1)])
|