Spaces:
Runtime error
Runtime error
File size: 323 Bytes
5395ccf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from .mask_rcnn_R_101_FPN_100ep_LSJ import (
dataloader,
lr_multiplier,
model,
optimizer,
train,
)
train.max_iter *= 4 # 100ep -> 400ep
lr_multiplier.scheduler.milestones = [
milestone * 4 for milestone in lr_multiplier.scheduler.milestones
]
lr_multiplier.scheduler.num_updates = train.max_iter
|