modify repsoftmax