experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #1197 from mozilla/decoder-load-tf1.5

Browse Source 
Move decoder module initialization to initialize_globals
This commit is contained in:
Reuben Morais 2018-01-24 13:16:46 +00:00 committed by GitHub
commit f9028aee1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
DeepSpeech.py
View File

@ -325,6 +325,14 @@ def initialize_globals():
log_error('Path specified in --one_shot_infer is not a valid file.') log_error('Path specified in --one_shot_infer is not a valid file.')
exit(1) exit(1)
if not os.path.exists(os.path.abspath(FLAGS.decoder_library_path)):
print('ERROR: The decoder library file does not exist. Make sure you have ' \
'downloaded or built the native client binaries and pass the ' \
'appropriate path to the binaries in the --decoder_library_path parameter.')
global custom_op_module
custom_op_module = tf.load_op_library(FLAGS.decoder_library_path)
# Logging functions # Logging functions
# ================= # =================
@ -478,13 +486,6 @@ def BiRNN(batch_x, seq_length, dropout):
# Output shape: [n_steps, batch_size, n_hidden_6] # Output shape: [n_steps, batch_size, n_hidden_6]
return layer_6 return layer_6
if not os.path.exists(os.path.abspath(FLAGS.decoder_library_path)):
print('ERROR: The decoder library file does not exist. Make sure you have ' \
'downloaded or built the native client binaries and pass the ' \
'appropriate path to the binaries in the --decoder_library_path parameter.')
custom_op_module = tf.load_op_library(FLAGS.decoder_library_path)
def decode_with_lm(inputs, sequence_length, beam_width=100, def decode_with_lm(inputs, sequence_length, beam_width=100,
top_paths=1, merge_repeated=True): top_paths=1, merge_repeated=True):
decoded_ixs, decoded_vals, decoded_shapes, log_probabilities = ( decoded_ixs, decoded_vals, decoded_shapes, log_probabilities = (