This commit is contained in:
Eren Golge 2019-10-24 14:34:31 +02:00
Родитель ea32f2368d
Коммит 9d5a5b0764
1 изменённых файлов: 1 добавлений и 0 удалений

Просмотреть файл

@ -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