Keras: как получить доступ к шагу в модели

Question

Есть ли способ получить доступ к текущему шагу обучения Keras в качестве тензора на графике тензорного потока?

Я пытаюсь построить модель с параметром 'epsilon', которыйзатухает в зависимости от текущего этапа обучения.

epsilon = some_fn_of(K.global_step) # <- Something like this?
self.q = K.Sequential([
    K.layers.InputLayer(input_shape),
    K.layers.Dense(n, name='q'),
    K.layers.Lambda(lambda x: tf.cond(tf.random.uniform((), 0, 1) < epsilon,
        lambda _: tf.constant(0.0),
        lambda ac: ac)
], name='q')

К вашему сведению: я использую комплект Keras с Tensorflow.

Answer 1

Я подозреваю, что у Keras нет такого тензора на графике и что единственный способ получить доступ к шагу - это обратные вызовы ( Keras Docs , Tensorflow Docs). Тем более, что Keras должен быть агностиком по отношению к бэкэнду, и поэтому, вероятно, будет поддерживать шаг за пределы тензорного потока.