Commit 7802e6f7 by xuchen

use the multiple configure file for training

parent c4748e46
......@@ -435,7 +435,7 @@ class S2TTransformerEncoder(FairseqEncoder):
self.history = None
self.use_ctc = "sate" in args.arch or \
(("ctc" in getattr(args, "criterion", False)) and \
(("ctc" in getattr(args, "criterion", False)) and
(getattr(args, "ctc_weight", False) > 0))
if self.use_ctc:
if task.source_dictionary == task.target_dictionary and getattr(args, "share_all_embeddings", False):
......
......@@ -106,6 +106,16 @@ def parse_args_and_arch(
is_config_file=True,
help="Configuration YAML filename (for training)",
)
parser.add_argument(
"--train-config3",
is_config_file=True,
help="Configuration YAML filename (for training)",
)
parser.add_argument(
"--train-config4",
is_config_file=True,
help="Configuration YAML filename (for training)",
)
if suppress_defaults:
# Parse args without any default values. This requires us to parse
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论