File size: 206 Bytes
e8f2571
 
 
 
 
 
 
1
2
3
4
5
6
7
8
_base_ = './detr_r50_8xb2-500e_coco.py'

model = dict(
    backbone=dict(
        depth=18,
        init_cfg=dict(type='Pretrained', checkpoint='torchvision://resnet18')),
    neck=dict(in_channels=[512]))