from local

This commit is contained in:
dohe0342 2023-01-24 16:28:35 +09:00
parent 3fb6e18159
commit 6d4c3d6165
2 changed files with 9 additions and 0 deletions

View File

@ -66,6 +66,15 @@ def get_parser():
parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter
)
parser.add_argument(
"--group-num",
type=int,
default=77,
help="""It specifies the checkpoint to use for decoding.
Note: Epoch counts from 1.
You can specify --avg to use more checkpoints for model averaging.""",
)
parser.add_argument(
"--epoch",