Spaces:
Running
on
T4
Running
on
T4
YOLO-World
/
third_party
/mmyolo
/configs
/razor
/subnets
/yolov5_s_spos_shufflenetv2_syncbn_8xb16-300e_coco.py
_base_ = [ | |
'mmrazor::_base_/nas_backbones/spos_shufflenet_supernet.py', | |
'../../yolov5/yolov5_s-v61_syncbn_fast_8xb16-300e_coco.py' | |
] | |
checkpoint_file = 'https://download.openmmlab.com/mmrazor/v1/spos/spos_shufflenetv2_subnet_8xb128_in1k_flops_0.33M_acc_73.87_20211222-1f0a0b4d_v3.pth' # noqa | |
fix_subnet = 'https://download.openmmlab.com/mmrazor/v1/spos/spos_shufflenetv2_subnet_8xb128_in1k_flops_0.33M_acc_73.87_20211222-1f0a0b4d_subnet_cfg_v3.yaml' # noqa | |
widen_factor = 1.0 | |
channels = [160, 320, 640] | |
_base_.nas_backbone.out_indices = (1, 2, 3) | |
_base_.nas_backbone.init_cfg = dict( | |
type='Pretrained', | |
checkpoint=checkpoint_file, | |
prefix='architecture.backbone.') | |
nas_backbone = dict( | |
type='mmrazor.sub_model', | |
fix_subnet=fix_subnet, | |
cfg=_base_.nas_backbone, | |
extra_prefix='architecture.backbone.') | |
_base_.model.backbone = nas_backbone | |
_base_.model.neck.widen_factor = widen_factor | |
_base_.model.neck.in_channels = channels | |
_base_.model.neck.out_channels = channels | |
_base_.model.bbox_head.head_module.in_channels = channels | |
_base_.model.bbox_head.head_module.widen_factor = widen_factor | |
find_unused_parameters = True | |