Use dropout when training softmax layer