如何在Keras训练期间记录张量的值？

Question

我在Keras中有一个定制的层，我想在训练期间监视特定的张量（即，每个时期评估该张量，并将值记录到列表中）。 我该怎么做才能做到这一点？

以一段代码为例：

class customizedLayer(Layer):
    def __init__():
        ...
    def build():
        ...
    def call():
        tensor_1 = ...
        tensor_2 = ...
        return ...

output = customizedLayer(input)
model = Model(input=X, output=Y)
model.compile(...)
model.fit(...)

如何获得上面显示的tensor_1或tensor_2的值？

非常感谢。

Answer 1

您可以实现一个Callback对象，以将每个我认为的值保存到文件中。 如果我有更具体的事情，我要告诉你。