Revert rnnt script to use standard label conversion

nv-rborkar · Aug 6, 2020 · 0cf2c57 · 0cf2c57
1 parent 08fba19
commit 0cf2c57
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/v0.7/speech_recognition/rnnt/accuracy_eval.py b/v0.7/speech_recognition/rnnt/accuracy_eval.py
@@ -41,11 +41,8 @@ def main():
         hypotheses.append(array.array(dtype_map[args.output_dtype], bytes.fromhex(result["data"])).tolist())
         references.append(manifest[result["qsl_idx"]]["transcript"])
 
-    # Convert ASCII output into string
-    for idx in range(len(hypotheses)):
-        hypotheses[idx] = ''.join([chr(c) for c in hypotheses[idx]])
-
     references = __gather_predictions([references], labels=labels)
+    hypotheses = __gather_predictions([hypotheses], labels=labels)
 
     d = dict(predictions=hypotheses,
              transcripts=references)