You could try to plot the custom_loss during training or plot all the three parts of loss separately and then observe the whole loss and the separate ones. I think this is a proper way to check whether your custom loss works.