mmrazor/tests/data/test_registry/registry_subnet_config.py

14 lines
282 B
Python
Raw Normal View History

# Copyright (c) OpenMMLab. All rights reserved.
supernet = dict(
type='MockModel',
)
model = dict(
type='MockAlgorithm',
architecture=supernet,
2022-07-14 14:38:35 +00:00
_fix_subnet_ = {
'architecture.mutable1': 'conv1',
'architecture.mutable2': 'conv2',
2022-07-14 14:38:35 +00:00
}
)