diff --git a/training/coqui_stt_training/train.py b/training/coqui_stt_training/train.py index 3c3821bb..8d563014 100644 --- 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