Fix float32 precision problem with high number of training