зеркало из https://github.com/mozilla/DeepSpeech.git
Expose some internal layers for downstream applications
This commit is contained in:
Родитель
7450e5763b
Коммит
d9bf55c39f
|
@ -814,6 +814,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
|
||||
|
|
Загрузка…
Ссылка в новой задаче