Learning to take losses