File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 88from mmengine .config import Config
99
1010# Add tools path so we can import the module directly
11- sys .path .insert (0 , 'tools/analysis_tools' ) # pragma: no cover
12- from get_flops_grounding import (
13- _get_backbone_out_channels , # noqa: E402
14- _get_feature_strides ,
15- _get_neck_out_channels ,
16- _get_transformer_config ,
17- count_neck_flops ,
18- count_text_encoder_flops ,
19- count_transformer_flops ,
20- format_flops ,
21- format_params )
11+ sys .path .insert (0 , 'tools/analysis_tools' )
12+ _mod = __import__ ( 'get_flops_grounding' )
13+ _get_backbone_out_channels = _mod . _get_backbone_out_channels
14+ _get_feature_strides = _mod . _get_feature_strides
15+ _get_neck_out_channels = _mod . _get_neck_out_channels
16+ _get_transformer_config = _mod . _get_transformer_config
17+ count_neck_flops = _mod . count_neck_flops
18+ count_text_encoder_flops = _mod . count_text_encoder_flops
19+ count_transformer_flops = _mod . count_transformer_flops
20+ format_flops = _mod . format_flops
21+ format_params = _mod . format_params
2222
2323
2424class TestFormatFlops (unittest .TestCase ):
You can’t perform that action at this time.
0 commit comments