Update train2.py

This commit is contained in:
jinzr 2023-10-11 09:55:50 +08:00
parent 7c37a8f933
commit 87d71af2ce

View File

@ -257,6 +257,16 @@ def get_parser():
""",
)
parser.add_argument(
"--lang-dir",
type=str,
default="data/lang_char",
help="""The lang dir
It contains language related input files such as
"lexicon.txt"
""",
)
parser.add_argument(
"--base-lr", type=float, default=0.05, help="The base learning rate."
)