Commit bec87003 by xuchen

use ctc when arch is sate

parent b8741ec8
......@@ -356,8 +356,9 @@ class S2TTransformerEncoder(FairseqEncoder):
else:
self.layer_norm = None
self.use_ctc = ("ctc" in getattr(args, "criterion", False)) and \
(getattr(args, "ctc_weight", False) > 0)
self.use_ctc = "sate" in args.arch or \
(("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):
self.ctc_projection = nn.Linear(
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论