Error in train (different losts)