add sgd for training.