diff --git a/layers/losses.py b/layers/losses.py index ab472519..79f5b381 100644 --- a/layers/losses.py +++ b/layers/losses.py @@ -58,6 +58,7 @@ class MSELossMasked(nn.Module): class AttentionEntropyLoss(nn.Module): + # pylint: disable=R0201 def forward(self, align): """ Forces attention to be more decisive by penalizing