From 4789bd1a4f5f2e960aefe376473309ca31b5ccbd Mon Sep 17 00:00:00 2001
From: Kayce001 <83959004+Kayce001@users.noreply.github.com>
Date: Sat, 6 Aug 2022 11:39:28 +0800
Subject: [PATCH] Update keypoint.ipynb (#292)

* Update keypoint.ipynb

add "with torch.no_grad():" in line 60.

* Update keypoint.ipynb
---
 tools/keypoint.ipynb | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tools/keypoint.ipynb b/tools/keypoint.ipynb
index 82a9aa4d..38127337 100644
--- a/tools/keypoint.ipynb
+++ b/tools/keypoint.ipynb
@@ -57,7 +57,8 @@
    "outputs": [],
    "source": [
     "output = non_max_suppression_kpt(output, 0.25, 0.65, nc=model.yaml['nc'], nkpt=model.yaml['nkpt'], kpt_label=True)\n",
-    "output = output_to_keypoint(output)\n",
+    "with torch.no_grad():\n",
+    "    output = output_to_keypoint(output)\n",
     "nimg = image[0].permute(1, 2, 0) * 255\n",
     "nimg = nimg.cpu().numpy().astype(np.uint8)\n",
     "nimg = cv2.cvtColor(nimg, cv2.COLOR_RGB2BGR)\n",