1 files changed, 1 insertions, 1 deletions
diff --git a/main.py b/main.py
index d384edb..b71d339 100644
--- a/main.py
+++ b/main.py
@@ -92,7 +92,7 @@ y_hat.name = 'y_hat'
 y_hat_mask = x_mask
 
 # Cost
-cost = CTC().apply(y, y_hat, y_mask, y_hat_mask)
+cost = CTC().apply(y, y_hat, y_mask.sum(axis=1), y_hat_mask).mean()
 cost.name = 'CTC'
 
 # Initialization