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

Expose some internal layers for downstream applications

This commit is contained in:
CatalinVoss 2021-03-03 12:55:25 -08:00 committed by Reuben Morais
parent 653ce25a7c
commit 900a01305c
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
training/coqui_stt_training/train.py
View File

@ -769,6 +769,10 @@ def create_inference_graph(batch_size=1, n_steps=16, tflite=False):
'new_state_c': new_state_c, 'new_state_c': new_state_c,
'new_state_h': new_state_h, 'new_state_h': new_state_h,
'mfccs': mfccs, 'mfccs': mfccs,
# Expose internal layers for downstream applications
'layer_3': layers['layer_3'],
'layer_5': layers['layer_5']
} }
return inputs, outputs, layers return inputs, outputs, layers