From c2f336b3e31640ab283e3231e2ea4c1ac44ace43 Mon Sep 17 00:00:00 2001
From: Reuben Morais <reuben.morais@gmail.com>
Date: Wed, 12 Feb 2020 13:08:34 +0100
Subject: [PATCH] Remove hardcoded constants from evaluate_tflite.py

---
 evaluate_tflite.py | 7 -------
 1 file changed, 7 deletions(-)

diff --git a/evaluate_tflite.py b/evaluate_tflite.py
index d01db864..f105702f 100644
--- a/evaluate_tflite.py
+++ b/evaluate_tflite.py
@@ -30,16 +30,9 @@ This module should be self-contained:
 Then run with a TF Lite model, a scorer and a CSV test file
 '''
 
-BEAM_WIDTH = 500
-LM_ALPHA = 0.75
-LM_BETA = 1.85
-
 def tflite_worker(model, scorer, queue_in, queue_out, gpu_mask):
     os.environ['CUDA_VISIBLE_DEVICES'] = str(gpu_mask)
     ds = Model(model)
-    ds.setBeamWidth(BEAM_WIDTH)
-    ds.enableExternalScorer(scorer)
-    ds.setScorerAlphaBeta(LM_ALPHA, LM_BETA)
 
     while True:
         try: