add SoftMaxTrain