fixing GPU management

openvinotoolkit · Mar 5, 2024 · 5f3e7eb · 5f3e7eb
1 parent 1920df4
commit 5f3e7eb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/230-yolov8-optimization/230-yolov8-object-detection.ipynb b/notebooks/230-yolov8-optimization/230-yolov8-object-detection.ipynb
@@ -1152,8 +1152,8 @@
     "if device.value != \"CPU\":\n",
     "    quantized_det_model.reshape({0: [1, 3, 640, 640]})\n",
     "if \"GPU\" in device.value or (\"AUTO\" in device.value and \"GPU\" in core.available_devices):\n",
-    "    ov_config = {\"GPU_DISABLE_WINOGRAD_CONVOLUTION\": \"YES\"}\n",
-    "quantized_det_compiled_model = core.compile_model(quantized_det_model, device.value, ov_config {\"GPU_DISABLE_WINOGRAD_CONVOLUTION\": \"YES\"} )\n",
+    "    ov_config = {}\n",
+    "quantized_det_compiled_model = core.compile_model(quantized_det_model, device.value)\n",
     "input_image = np.array(Image.open(IMAGE_PATH))\n",
     "detections = detect(input_image, quantized_det_compiled_model)[0]\n",
     "image_with_boxes = draw_results(detections, input_image, label_map)\n",