Expose some internal layers for downstream applications

2021-03-03 12:55:25 -08:00 · 2021-03-03 12:55:25 -08:00 · 900a01305c
parent 653ce25a7c
commit 900a01305c
1 changed files with 4 additions and 0 deletions
--- a/training/coqui_stt_training/train.py
+++ b/training/coqui_stt_training/train.py
@ -769,6 +769,10 @@ def create_inference_graph(batch_size=1, n_steps=16, tflite=False):
        'new_state_c': new_state_c,
        'new_state_h': new_state_h,
        'mfccs': mfccs,
+
+        # Expose internal layers for downstream applications
+        'layer_3': layers['layer_3'],
+        'layer_5': layers['layer_5']
    }

    return inputs, outputs, layers