Commit 7802e6f7 by xuchen

use the multiple configure file for training

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