Now we train all points every epoch