transfiner / configs /new_baselines /mask_rcnn_R_50_FPN_50ep_LSJ.py
lkeab
update configs
c6c496f
from .mask_rcnn_R_50_FPN_100ep_LSJ import (
dataloader,
lr_multiplier,
model,
optimizer,
train,
)
train.max_iter //= 2 # 100ep -> 50ep
lr_multiplier.scheduler.milestones = [
milestone // 2 for milestone in lr_multiplier.scheduler.milestones
]
lr_multiplier.scheduler.num_updates = train.max_iter