Обратный вызов Keras для вывода промежуточного плотного слоя

Question

Я пытаюсь реализовать обратный вызов, который в конце обучения сохраняет вывод промежуточного плотного слоя в файл json. Моя попытка создает файл и не выдает ошибку, но файл пуст. Как мне поступить об этом?

import json
json_log = open('fingerprints.json', mode='wt', buffering=1)
fingerprint_callback = tf.keras.callbacks.LambdaCallback(
    on_train_end=lambda _: json_log.write(
        json.dumps({'fingerprints': model.layers[-5].output})))