Номер текущей эпохи в тензорном потоке настраиваемого слоя python

Question

Я впервые задаю вопрос, так что go мне легко. Я пытаюсь написать собственный слой в python tensorflow / keras, и в слое я хочу получить доступ к номеру текущей эпохи модели. Я тщательно искал документацию по тензорному потоку, чтобы найти что-то, что могло бы дать мне это. Лучшее, что я смог найти, - это tensorflow.keras.backend.learning_phase(), которые возвращаются независимо от того, является ли это фазой обучения или нет. Пожалуйста, помогите!

Answer 1

Вам необходимо создать обратный вызов, проверьте документацию keras

from tensorflow.keras.callbacks import Callback


class CustomEpoch(Callback):
    def check_condition(self, epoch):
        # TODO check if the condition is met
        pass

    def on_epoch_begin(self, epoch, logs=None):
        if not self.check_condition(epoch):
            return
        # TODO do stuff

и убедитесь, что это где-то вызывается:

model.fit(callbacks=[CustomEpoch()])