From 94db28e088319bf44bb8f4f11b232e6ace9c0300 Mon Sep 17 00:00:00 2001
From: Xinghai Sun <sunxinghai1216@gmail.com>
Date: Thu, 3 Aug 2017 21:06:27 +0800
Subject: [PATCH] Add function doc for infer_batch_loss() function in model.py
 for DS2.

---
 model.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/model.py b/model.py
index c8766deb..c2e440b3 100644
--- a/model.py
+++ b/model.py
@@ -120,6 +120,16 @@ class DeepSpeech2Model(object):
             feeding=feeding_dict)
 
     def infer_loss_batch(self, infer_data):
+        """Model inference. Infer the ctc loss for a batch of speech
+        utterances.
+
+        :param infer_data: List of utterances to infer, with each utterance a
+                           tuple of audio features and transcription text (empty
+                           string).
+        :type infer_data: list
+        :return: List of ctc loss.
+        :rtype: List of float
+        """
         # define inferer
         if self._loss_inferer == None:
             self._loss_inferer = paddle.inference.Inference(