use sigmoid

Summary: use sigmoid

Reviewed By: liaimi

Differential Revision: D19352122

fbshipit-source-id: 2640f554221d56d9fe2c8f0a9874f31ea0159d58

Author: chenyangyu1988

pytext/models/output_layers/doc_classification_output_layer.py

@@ -183,7 +183,7 @@ def get_pred(self, logit, *args, **kwargs):
         return preds, scores
 
     def torchscript_predictions(self):
-        return ClassificationScores(self.target_names, F.log_softmax)
+        return ClassificationScores(self.target_names, F.logsigmoid)
 
     def export_to_caffe2(
         self,