update trainer

weakly_supervised_parser/model/trainer.py

@@ -99,8 +99,8 @@ class InsideOutsideStringClassifier:
 
     def load_model(self, pre_trained_model_path):
         options = SessionOptions()
-        options.intra_op_num_threads = 16
-        options.inter_op_num_threads = 16
+        options.intra_op_num_threads = 32
+        options.inter_op_num_threads = 32
         self.model = InferenceSession(pre_trained_model_path, options, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path, use_fast=True)