Skip to content

Commit 2967e8c

Browse files
committed
fix dep again
1 parent a91e397 commit 2967e8c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -276,7 +276,7 @@ def sequence_mask(sequence_length, max_len=None):
276276
class MaskedL1Loss(nn.Module):
277277
def __init__(self):
278278
super(MaskedL1Loss, self).__init__()
279-
self.criterion = nn.L1Loss(size_average=False)
279+
self.criterion = nn.L1Loss(reduction="sum")
280280

281281
def forward(self, input, target, lengths=None, mask=None, max_len=None):
282282
if lengths is None and mask is None:

0 commit comments

Comments
 (0)