This commit is contained in:
Eren Golge
2019-10-24 14:19:16 +02:00
parent ea32f2368d
commit 3b7aa67ed2
+1 -1
View File
@@ -58,7 +58,7 @@ class MSELossMasked(nn.Module):
class AttentionEntropyLoss(nn.Module):
def forward(self, align):
def forward(self, align): #pylint disable=no-self-use
"""
Forces attention to be more decisive by penalizing
soft attention weights