fix moniroting for nlp softmax