Fix use of export_language flag

This commit is contained in:
Reuben Morais 2019-04-05 21:37:05 -03:00
parent 5779d298e1
commit d08fc4b6a2

View File

@ -718,8 +718,8 @@ def export():
metadata.attr['sample_rate'].i = FLAGS.audio_sample_rate
metadata.attr['feature_win_len'].i = FLAGS.feature_win_len
metadata.attr['feature_win_step'].i = FLAGS.feature_win_step
if FLAGS.export_model_language:
metadata.attr['language'].s = FLAGS.export_model_language.encode('ascii')
if FLAGS.export_language:
metadata.attr['language'].s = FLAGS.export_language.encode('ascii')
with open(output_graph_path, 'wb') as fout:
fout.write(frozen_graph.SerializeToString())