Update get_flops.py (#1371)

* Update get_flops.py argparse.ArgumentParser(description='Get the FLOPs of a segmentor') * Update get_flops.py
open-mmlab · Mar 14, 2022 · a3aab38 · a3aab38
1 parent 17b500f
commit a3aab38
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',