python - Q : Best way to extract model output during training with Keras

Question

正如问题所说，我想在训练期间为每个输入数据提取模型输出。我认为在训练期间可以轻松提取前向传递的输出，因为它无论如何都是计算出来的，但无法找到一种有效的方法来做到这一点。

所以我正在寻找一种更有效的方法来执行以下操作:

for epoch in range(num_epochs):
    model.fit(x_tr, y_tr, batch_size=batch_size, epochs=1)
    y_pred = model.predict(x_tr, batch_size=batch_size)

    # now I use y_pred for something before starting the next epoch
    do_something_with_prediction(y_pred)

Custom Callbacks进入我的脑海，因为它们具有可以调用的内置函数，例如在每批结束时。但据我所知，您只能访问损失/度量值，而不能访问最后一层的实际输出。

好吧，我认为必须有一种更有效的方法，因为输出是在每次向前-向后传递时计算的，我真的很想节省时间来再次计算整个训练数据的正向传递。

Answer 1

你可以在图形模式下以高性能做这样的事情，

tf.py_function(do_something_with_prediction, [y_pred], [outputs_of_your_func])