Update get_flops.py (open-mmlab#1371)

* Update get_flops.py argparse.ArgumentParser(description='Get the FLOPs of a segmentor') * Update get_flops.py
AetrexTechnology · Mar 14, 2022 · 912d109 · 912d109
1 parent 42df28c
commit 912d109
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tools/get_flops.py b/tools/get_flops.py
@@ -8,7 +8,8 @@
 
 
 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',