Visualizing the losses