Commit afa5095d by xuchen

fix the bugs of mixup

parent c845197f
......@@ -899,10 +899,11 @@ class S2TTransformerEncoder(FairseqEncoder):
layer_idx = -1
mixup = None
if type(self.mixup_layer) is list:
mixup_layer = choice(self.mixup_layer)
else:
mixup_layer = self.mixup_layer
if self.mixup:
if type(self.mixup_layer) is list:
mixup_layer = choice(self.mixup_layer)
else:
mixup_layer = self.mixup_layer
if self.history is not None:
self.history.clean()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论