Update get_flops.py (#1371)

* Update get_flops.py

argparse.ArgumentParser(description='Get the FLOPs of a segmentor')

* Update get_flops.py
pull/1378/head^2
wangliujun 2022-03-14 21:02:43 +08:00 committed by GitHub
parent 17b500f9df
commit a3aab38bb4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -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',