Fixed log loss.

geneing · Jul 2, 2020 · 0e4f6a3 · 0e4f6a3
1 parent 88bb65e
commit 0e4f6a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trainer/common.py b/trainer/common.py
@@ -74,7 +74,7 @@ def forward(self, x, target, lens):
 
 class LogL1(torch.nn.Module):
 
-    def forward(self, x, target, lens):
+    def forward(self, x, target):
         target.requires_grad = False
         target = torch.log(target + 1.0)
         x = torch.log(x + 1.0)