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 object，以将每个值保存到文件中。如果我有更具体的事情，我会告诉你。

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

1 个答案: