From 2087b63a37f8a6fa3bc1829ef11c4af312ba62d6 Mon Sep 17 00:00:00 2001 From: Kevin Wang Date: Tue, 11 Jul 2023 17:47:22 +0800 Subject: [PATCH] [Fix] fix an error when using get_flops.py script to get textrecog model flops --- tools/analysis_tools/get_flops.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/analysis_tools/get_flops.py b/tools/analysis_tools/get_flops.py index caa97203a..092520f47 100644 --- a/tools/analysis_tools/get_flops.py +++ b/tools/analysis_tools/get_flops.py @@ -38,6 +38,7 @@ def main(): cfg = Config.fromfile(args.config) init_default_scope(cfg.get('default_scope', 'mmocr')) model = MODELS.build(cfg.model) + model.eval() flops = FlopCountAnalysis(model, torch.ones(input_shape))