Expose some internal layers for downstream applications

This commit is contained in:
CatalinVoss 2021-03-03 12:55:25 -08:00
Родитель 7450e5763b
Коммит d9bf55c39f
1 изменённых файлов: 4 добавлений и 0 удалений

Просмотреть файл

@ -814,6 +814,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