时间戳

首页 > 解决方案 > 自定义层张量流python中的当前纪元数

python - 自定义层张量流python中的当前纪元数

问题描述

这是我第一次问问题，所以请放轻松。我正在尝试在 python tensorflow/keras 中编写一个自定义层，并且在我想访问模型当前时代的层中。我已经广泛搜索了 tensorflow 文档，以找到可以给我的东西。我能找到的最好的结果是tensorflow.keras.backend.learning_phase()无论是否是学习阶段。请帮忙！

标签： pythontensorflowkeras

解决方案

你需要创建一个回调，查看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()])

推荐阅读