Training on the full dataset