charlie880624/STDC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update get_flops.py (#1371)

Browse Source 
* Update get_flops.py

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

* Update get_flops.py
This commit is contained in:
wangliujun 2022-03-14 21:02:43 +08:00 committed by EricWu
parent 43a6c0d5d7
commit 08ee236106
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/get_flops.py
View File

@ -8,7 +8,8 @@ from mmseg.models import build_segmentor
def parse_args(): 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('config', help='train config file path')
parser.add_argument( parser.add_argument(
'--shape', '--shape',