Update train.py

modified a default param.
This commit is contained in:
zr_jin 2024-03-07 16:35:41 +08:00 committed by GitHub
parent 19e21ba3ff
commit 1c6a6a2011
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -135,7 +135,7 @@ def get_parser():
parser.add_argument(
"--exp-dir",
type=str,
default="pruned_transducer_stateless7/exp",
default="whisper/exp",
help="""The experiment dir.
It specifies the directory where all training related
files, e.g., checkpoints, log, etc, are saved