Skip to content

Commit 5170483

Browse files
committed
fix bug
Signed-off-by: Xin He <[email protected]>
1 parent 958d3b0 commit 5170483

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

test/export/test_torch2onnx.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -94,12 +94,12 @@ def setUpClass(self):
9494
@classmethod
9595
def tearDownClass(self):
9696
shutil.rmtree('nc_workspace', ignore_errors=True)
97-
# os.remove('fp32-cv-model.onnx')
98-
# os.remove('int8-cv-qdq-model.onnx')
99-
# os.remove('int8-cv-qlinear-model.onnx')
100-
# os.remove('fp32-nlp-model.onnx')
101-
# os.remove('int8-nlp-qdq-model.onnx')
102-
# os.remove('int8-nlp-qlinear-model.onnx')
97+
os.remove('fp32-cv-model.onnx')
98+
os.remove('int8-cv-qdq-model.onnx')
99+
os.remove('int8-cv-qlinear-model.onnx')
100+
os.remove('fp32-nlp-model.onnx')
101+
os.remove('int8-nlp-qdq-model.onnx')
102+
os.remove('int8-nlp-qlinear-model.onnx')
103103

104104
def test_fp32_CV_models(self):
105105
model = self.cv_model

0 commit comments

Comments
 (0)