Estou tentando criar uma função de perda no keras (Tensorflow Backend), mas estou um pouco paralisado para verificar o interior da função de perda personalizada. De fato, a impressão aparece no console somente quando eu compilo o modelo, depois disso não há impressão. (Estou apenas testando uma função personalizada muito simples, criarei a função verdadeira quando resolver esse problema). Treino o modelo usando a função train_on_batch. Como posso resolver este problema?

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))

minha saída:

função de perda interna

treinar em lote

-0,481604