From a1394aad6fca2dd560eb45a9b2e4cbc7be4c2bf7 Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Wed, 23 Dec 2015 18:35:56 +0100 Subject: stuff --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'main.py') 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 -- cgit v1.2.3