Fixed invalid feature vector size for training.