Блог

Я хочу определить функцию потерь, и потеря 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