File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11# Copyright (c) OpenMMLab. All rights reserved.
2- """Get FLOPs for GroundingDINO models.
2+ """Get FLOPs for GroundingDINO/GroundingCLIP models.
33
44This script is specifically designed for models that require text inputs,
5- such as GroundingDINO and other vision-language models.
5+ such as GroundingDINO, GroundingCLIP, and other vision-language models.
66
77Gradient Checkpointing (with_cp) Compatibility:
88 This script uses fvcore's FlopCountAnalysis for accurate backbone
@@ -438,7 +438,7 @@ def main():
438438
439439 split_line = '=' * 60
440440 logger .info (split_line )
441- logger .info ('GroundingDINO FLOPs Analysis' )
441+ logger .info ('GroundingDINO/GroundingCLIP FLOPs Analysis' )
442442 logger .info (split_line )
443443 logger .info (f'Config: { args .config } ' )
444444 logger .info (f'Input shape: ({ h } , { w } )' )
You can’t perform that action at this time.
0 commit comments