Update whitebox_utils/classifier.py

whitebox_utils/classifier.py

@@ -112,6 +112,7 @@ class MyClassifier():
             batch = self.tokenize(sent,idx=self.count)
             inputs = {"input_ids": batch[0], "attention_mask": batch[1], "token_type_ids":batch[2]}
             outputs = self.model(**inputs)[0] # get all logits
+            print(outputs)
             output_ = [torch.softmax(output,dim=1)[0].detach().cpu().numpy() for output in outputs]
             ret.append(output_)
         return np.array(ret)