Update text/g2pw/onnx_api.py
Browse files- text/g2pw/onnx_api.py +1 -4
text/g2pw/onnx_api.py
CHANGED
@@ -90,10 +90,7 @@ class G2PWOnnxConverter:
|
|
90 |
print(":::4")
|
91 |
sess_options.intra_op_num_threads = 2
|
92 |
print(":::5")
|
93 |
-
|
94 |
-
self.session_g2pW = onnxruntime.InferenceSession(os.path.join(uncompress_path, 'g2pW.onnx'),sess_options=sess_options, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
|
95 |
-
except:
|
96 |
-
self.session_g2pW = onnxruntime.InferenceSession(os.path.join(uncompress_path, 'g2pW.onnx'),sess_options=sess_options, providers=['CPUExecutionProvider'])
|
97 |
print(":::6")
|
98 |
self.config = load_config(
|
99 |
config_path=os.path.join(uncompress_path, 'config.py'),
|
|
|
90 |
print(":::4")
|
91 |
sess_options.intra_op_num_threads = 2
|
92 |
print(":::5")
|
93 |
+
self.session_g2pW = onnxruntime.InferenceSession(os.path.join(uncompress_path, 'g2pW.onnx'),sess_options=sess_options, providers=['CPUExecutionProvider'])
|
|
|
|
|
|
|
94 |
print(":::6")
|
95 |
self.config = load_config(
|
96 |
config_path=os.path.join(uncompress_path, 'config.py'),
|