| 12345678910111213141516171819202122232425262728293031323334353637 |
- _base_ = './mask2former_r50_8xb2-8e_youtubevis2019.py'
- dataset_type = 'YouTubeVISDataset'
- data_root = 'data/youtube_vis_2021/'
- dataset_version = data_root[-5:-1] # 2019 or 2021
- train_dataloader = dict(
- dataset=dict(
- data_root=data_root,
- dataset_version=dataset_version,
- ann_file='annotations/youtube_vis_2021_train.json'))
- val_dataloader = dict(
- dataset=dict(
- data_root=data_root,
- dataset_version=dataset_version,
- ann_file='annotations/youtube_vis_2021_valid.json'))
- test_dataloader = val_dataloader
- # learning policy
- max_iters = 8000
- param_scheduler = dict(
- type='MultiStepLR',
- begin=0,
- end=max_iters,
- by_epoch=False,
- milestones=[
- 5500,
- ],
- gamma=0.1)
- # runtime settings
- train_cfg = dict(
- type='IterBasedTrainLoop', max_iters=max_iters, val_interval=8001)
- default_hooks = dict(
- checkpoint=dict(
- type='CheckpointHook', by_epoch=False, save_last=True, interval=500))
|
