mmrazor/tests/data/test_registry/registry_subnet_config.py
2022-07-15 23:05:07 +08:00

14 lines
282 B
Python

# Copyright (c) OpenMMLab. All rights reserved.
supernet = dict(
type='MockModel',
)
model = dict(
type='MockAlgorithm',
architecture=supernet,
_fix_subnet_ = {
'architecture.mutable1': 'conv1',
'architecture.mutable2': 'conv2',
}
)