Skip to content

Commit 157ce34

Browse files
committed
Update evaluate_tflite.py script for Coqpit
1 parent 2758403 commit 157ce34

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

evaluate_tflite.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,9 @@
1010
from functools import partial
1111
from multiprocessing import JoinableQueue, Manager, Process, cpu_count
1212

13-
import absl.app
1413
import numpy as np
1514
from coqui_stt_training.util.evaluate_tools import calculate_and_print_report
16-
from coqui_stt_training.util.flags import create_flags
15+
from coqui_stt_training.util.config import Config, initialize_globals_from_args
1716
from six.moves import range, zip
1817
from stt import Model
1918

@@ -61,6 +60,7 @@ def tflite_worker(model, scorer, queue_in, queue_out, gpu_mask):
6160

6261

6362
def main(args):
63+
initialize_globals_from_args()
6464
manager = Manager()
6565
work_todo = JoinableQueue() # this is where we are going to store input data
6666
work_done = manager.Queue() # this where we are gonna push them out

0 commit comments

Comments
 (0)