Skip to content

Commit a3aab38

Browse files
Update get_flops.py (open-mmlab#1371)
* Update get_flops.py argparse.ArgumentParser(description='Get the FLOPs of a segmentor') * Update get_flops.py
1 parent 17b500f commit a3aab38

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tools/get_flops.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,8 @@
88

99

1010
def parse_args():
11-
parser = argparse.ArgumentParser(description='Train a segmentor')
11+
parser = argparse.ArgumentParser(
12+
description='Get the FLOPs of a segmentor')
1213
parser.add_argument('config', help='train config file path')
1314
parser.add_argument(
1415
'--shape',

0 commit comments

Comments
 (0)