Как вызвать глобальную переменную в пользовательской функции потерь keras, чтобы изменить возвращаемое значение функции потерь

Question

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

def my_loss(y_true, y_pred):

    global loss_count
    loss_count += 1
    print("the loss",loss_count)
    return keras.losses.categorical_crossentropy(y_true, y_pred)

После обучения модели переменный счет потерь равен 1