|
|
|
@ -46,8 +46,10 @@ class G2PWOnnxConverter:
|
|
|
|
|
uncompress_path = download_and_decompress(g2pw_onnx_models['G2PWModel']['1.0'],model_dir)
|
|
|
|
|
|
|
|
|
|
sess_options = onnxruntime.SessionOptions()
|
|
|
|
|
sess_options.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL
|
|
|
|
|
sess_options.execution_mode = onnxruntime.ExecutionMode.ORT_SEQUENTIAL
|
|
|
|
|
sess_options.intra_op_num_threads = 2
|
|
|
|
|
self.session_g2pW = onnxruntime.InferenceSession(os.path.join(model_dir, 'G2PWModel/g2pW.onnx'))
|
|
|
|
|
self.session_g2pW = onnxruntime.InferenceSession(os.path.join(model_dir, 'G2PWModel/g2pW.onnx'),sess_options=sess_options)
|
|
|
|
|
self.config = load_config(os.path.join(model_dir, 'G2PWModel/config.py'), use_default=True)
|
|
|
|
|
|
|
|
|
|
self.model_source = model_source if model_source else self.config.model_source
|
|
|
|
|