File size: 707 Bytes
186701e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# Copyright (c) OpenMMLab. All rights reserved.
import copy
import pytest
from mmcls.models.backbones.base_backbone import BaseBackbone
from mmyolo.testing import get_detector_cfg
@pytest.mark.parametrize('cfg_file', [
'razor/subnets/'
'yolov5_s_spos_shufflenetv2_syncbn_8xb16-300e_coco.py', 'razor/subnets/'
'rtmdet_tiny_ofa_lat31_syncbn_16xb16-300e_coco.py', 'razor/subnets/'
'yolov6_l_attentivenas_a6_d12_syncbn_fast_8xb32-300e_coco.py'
])
def test_razor_backbone_init(cfg_file):
model = get_detector_cfg(cfg_file)
model_cfg = copy.deepcopy(model.backbone)
from mmrazor.registry import MODELS
model = MODELS.build(model_cfg)
assert isinstance(model, BaseBackbone)
|