Skip to content

Commit 2c870b3

Browse files
revert
1 parent 3e49e7a commit 2c870b3

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

tests/post_training/pipelines/image_classification_timm.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111

1212
import numpy as np
1313
import onnx
14-
import onnxoptimizer
1514
import openvino as ov
1615
import timm
1716
import torch
@@ -59,9 +58,7 @@ def prepare_model(self) -> None:
5958
**additional_kwargs,
6059
)
6160

62-
model = onnx.load(onnx_path)
63-
passes = ["fuse_bn_into_conv"]
64-
self.model = onnxoptimizer.optimize(model, passes)
61+
self.model = onnx.load(onnx_path)
6562
self.input_name = self.model.graph.input[0].name
6663

6764
if self.backend in OV_BACKENDS + [BackendType.FP32]:

tests/post_training/requirements.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,4 +23,3 @@ transformers==4.53.0
2323
whowhatbench @ git+https://github.com/openvinotoolkit/[email protected]#subdirectory=tools/who_what_benchmark
2424
datasets==3.6.0
2525
onnxscript
26-
onnxoptimizer==0.3.8

0 commit comments

Comments
 (0)