From 08ee236106f010db9779b2810d3162d8bff3b8e2 Mon Sep 17 00:00:00 2001 From: wangliujun Date: Mon, 14 Mar 2022 21:02:43 +0800 Subject: [PATCH] Update get_flops.py (#1371) * Update get_flops.py argparse.ArgumentParser(description='Get the FLOPs of a segmentor') * Update get_flops.py --- tools/get_flops.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/get_flops.py b/tools/get_flops.py index 83dea0a..e30c36f 100644 --- a/tools/get_flops.py +++ b/tools/get_flops.py @@ -8,7 +8,8 @@ from mmseg.models import build_segmentor def parse_args(): - parser = argparse.ArgumentParser(description='Train a segmentor') + parser = argparse.ArgumentParser( + description='Get the FLOPs of a segmentor') parser.add_argument('config', help='train config file path') parser.add_argument( '--shape',