Add missing #include of <limits> for loss functions. Change-Id: Id632451429e03031a1533a9be795270debc70706