| _base_ = [ | |
| '../_base_/models/mask-rcnn_r50_fpn.py', | |
| '../_base_/datasets/deepfashion.py', '../_base_/schedules/schedule_1x.py', | |
| '../_base_/default_runtime.py' | |
| ] | |
| model = dict( | |
| roi_head=dict( | |
| bbox_head=dict(num_classes=15), mask_head=dict(num_classes=15))) | |
| # runtime settings | |
| max_epochs = 15 | |
| train_cfg = dict( | |
| type='EpochBasedTrainLoop', max_epochs=max_epochs, val_interval=1) | |
| param_scheduler = [ | |
| dict( | |
| type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500), | |
| dict( | |
| type='MultiStepLR', | |
| begin=0, | |
| end=max_epochs, | |
| by_epoch=True, | |
| milestones=[8, 11], | |
| gamma=0.1) | |
| ] | |