train on union and add loss multiplier