ctc loss: if label prob is 0, then ctc loss is +inf (instead of throwing an exception)

githubharald · githubharald · commit 3a445949b880 · 2019-05-25T13:55:04.000+02:00
diff --git a/src/Loss.py b/src/Loss.py
@@ -58,7 +58,10 @@ def ctcLabelingProb(mat, gt, classes):
 
 def ctcLoss(mat, gt, classes):
 	"calculate CTC loss"
-	return -math.log(ctcLabelingProb(mat, gt, classes))
+	try:
+		return -math.log(ctcLabelingProb(mat, gt, classes))
+	except:
+		return float('inf')
 
 
 def testLoss():
