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})))