Update get_flops.py (#1371)
* Update get_flops.py argparse.ArgumentParser(description='Get the FLOPs of a segmentor') * Update get_flops.pypull/1378/head^2
parent
17b500f9df
commit
a3aab38bb4
|
@ -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',
|
||||
|
|
Loading…
Reference in New Issue