Add more losses to framework