Commit bec87003 by xuchen

use ctc when arch is sate

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