I have defined a custom loss function for my model.
def get_loss(y_hat, y): loss = tf.keras.losses.BinaryCrossentropy(y_hat,y) # cross entropy (but no logits