from local

This commit is contained in:
dohe0342 2022-12-20 16:17:48 +09:00
parent 33dd3e5141
commit ec079cc228
2 changed files with 8 additions and 0 deletions

View File

@ -85,6 +85,14 @@ def get_parser():
parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter
)
parser.add_argument(
"--model-name",
type=int,
default=30,
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",