From 157ce340b6c07c8201023e5cbd59dc165ef4211b Mon Sep 17 00:00:00 2001 From: Josh Meyer Date: Thu, 7 Oct 2021 14:46:03 -0400 Subject: [PATCH] Update evaluate_tflite.py script for Coqpit --- evaluate_tflite.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/evaluate_tflite.py b/evaluate_tflite.py index c075b4c4..baa33ae4 100644 --- a/evaluate_tflite.py +++ b/evaluate_tflite.py @@ -10,10 +10,9 @@ import wave from functools import partial from multiprocessing import JoinableQueue, Manager, Process, cpu_count -import absl.app import numpy as np from coqui_stt_training.util.evaluate_tools import calculate_and_print_report -from coqui_stt_training.util.flags import create_flags +from coqui_stt_training.util.config import Config, initialize_globals_from_args from six.moves import range, zip from stt import Model @@ -61,6 +60,7 @@ def tflite_worker(model, scorer, queue_in, queue_out, gpu_mask): def main(args): + initialize_globals_from_args() manager = Manager() work_todo = JoinableQueue() # this is where we are going to store input data work_done = manager.Queue() # this where we are gonna push them out