Spaces:
Runtime error
Runtime error
# Copyright (c) OpenMMLab. All rights reserved. | |
from mmengine.config import read_base | |
from mmengine.optim.scheduler.lr_scheduler import MultiStepLR | |
from mmengine.runner.loops import EpochBasedTrainLoop | |
with read_base(): | |
from .detr_r50_8xb2_150e_coco import * | |
# learning policy | |
max_epochs = 500 | |
train_cfg.update( | |
type=EpochBasedTrainLoop, max_epochs=max_epochs, val_interval=10) | |
param_scheduler = [ | |
dict( | |
type=MultiStepLR, | |
begin=0, | |
end=max_epochs, | |
by_epoch=True, | |
milestones=[334], | |
gamma=0.1) | |
] | |
# only keep latest 2 checkpoints | |
default_hooks.update(checkpoint=dict(max_keep_ckpts=2)) | |