This commit is contained in:
Eren Golge 2019-10-24 14:34:31 +02:00
parent ea32f2368d
commit 9d5a5b0764
1 changed files with 1 additions and 0 deletions

View File

@ -58,6 +58,7 @@ class MSELossMasked(nn.Module):
class AttentionEntropyLoss(nn.Module): class AttentionEntropyLoss(nn.Module):
# pylint: disable=R0201
def forward(self, align): def forward(self, align):
""" """
Forces attention to be more decisive by penalizing Forces attention to be more decisive by penalizing