small fix

This commit is contained in:
marcoyang 2023-02-13 16:21:21 +08:00
parent 6018f222df
commit 142ea421b0

View File

@ -157,6 +157,13 @@ def add_model_arguments(parser: argparse.ArgumentParser):
value of gradient norms of all elememts in batch.""", value of gradient norms of all elememts in batch.""",
) )
parser.add_argument(
"--is-pnnx",
type=str2bool,
default=False,
help="Only used when exporting model with pnnx."
)
def get_parser(): def get_parser():
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(