Блог

У меня есть настраиваемый слой в 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(...)
  

что я должен сделать, чтобы получить значения тензора_1 или тензора_2, показанные выше?

Большое спасибо.

Я думаю, вы можете реализовать объект обратного вызова для сохранения этих значений в файл в каждую эпоху. Если у меня есть что-то более конкретное, я говорю вам.