update trainer

packages.txt

@@ -1,6 +0,0 @@
-libcudart
-libcufft
-libcurand
-libcublasLt
-libcublas
-libcudnn

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 = 1
-        options.inter_op_num_threads = 1
+        options.intra_op_num_threads = 16
+        options.inter_op_num_threads = 16
         self.model = InferenceSession(pre_trained_model_path, options, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path, use_fast=True)