examples/benchmark fix for resnet50 example (#1597)

neuralmagic · Feb 12, 2024 · 20f90ac · 20f90ac
1 parent 517fd15
commit 20f90ac
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/benchmark/resnet50_benchmark.py b/examples/benchmark/resnet50_benchmark.py
@@ -47,7 +47,8 @@
 
 import numpy
 
-from deepsparse import benchmark_model, cpu
+from deepsparse import cpu
+from deepsparse.engine import benchmark_model
 
 
 CORES_PER_SOCKET, AVX_TYPE, VNNI = cpu.cpu_details()