pull/10/head
Eren Golge 2019-10-24 14:19:16 +02:00
parent ea32f2368d
commit 3b7aa67ed2
1 changed files with 1 additions and 1 deletions

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