аюсь создать функцию потерь в keras (Tensorflow Backend), но я немного застрял, чтобы проверить внутреннюю часть пользовательской функции потерь. На самом деле, печать появляется на консоли только тогда, когда я компилирую модель, после чего нет печати. (Я просто тестирую очень простую пользовательскую функцию, я создам настоящую функцию, когда решу эту проблему). Я тренирую модель, используя функцию train_on_batch. Как я могу решить эту проблему ?

def loss_yolo(self, y_true, y_pred):  
    '''
        4*7*7 = 196
        1*7*7 = 49
        7*7*20 = 980
    '''
    print('inside loss function')
    loss = K.mean(y_true- y_pred)
    return loss
model.compile(optimizer='sgd', loss=loss_yolo)
print('train on batch')
print(model.train_on_batch(x, y))

мой вывод:

функция внутренней потери

поезд на партии

-0,481604