Skip to content

Commit a15543e

Browse files
authored
[cherry-pick] examples/benchmark fix for resnet50 example (#1597) (#1598)
1 parent 30ba1ca commit a15543e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Diff for: examples/benchmark/resnet50_benchmark.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,8 @@
4747

4848
import numpy
4949

50-
from deepsparse import benchmark_model, cpu
50+
from deepsparse import cpu
51+
from deepsparse.engine import benchmark_model
5152

5253

5354
CORES_PER_SOCKET, AVX_TYPE, VNNI = cpu.cpu_details()

0 commit comments

Comments
 (0)