Update notebooks torch.hub.load() examples (#11952)

* Update tutorial.ipynb Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update tutorial.ipynb Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update tutorial.ipynb Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update tutorial.ipynb Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> --------- Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
ultralytics · Aug 8, 2023 · df48c20 · df48c20
1 parent 934d5d1
commit df48c20
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/classify/tutorial.ipynb b/classify/tutorial.ipynb
@@ -1445,7 +1445,7 @@
         "# YOLOv5 PyTorch HUB Inference (DetectionModels only)\n",
         "import torch\n",
         "\n",
-        "model = torch.hub.load('ultralytics/yolov5', 'yolov5s')  # yolov5n - yolov5x6 or custom\n",
+        "model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True, trust_repo=True)  # or yolov5n - yolov5x6 or custom\n",
         "im = 'https://ultralytics.com/images/zidane.jpg'  # file, Path, PIL.Image, OpenCV, nparray, list\n",
         "results = model(im)  # inference\n",
         "results.print()  # or .show(), .save(), .crop(), .pandas(), etc."

diff --git a/segment/tutorial.ipynb b/segment/tutorial.ipynb
@@ -558,7 +558,7 @@
         "# YOLOv5 PyTorch HUB Inference (DetectionModels only)\n",
         "import torch\n",
         "\n",
-        "model = torch.hub.load('ultralytics/yolov5', 'yolov5s-seg')  # yolov5n - yolov5x6 or custom\n",
+        "model = torch.hub.load('ultralytics/yolov5', 'yolov5s-seg', force_reload=True, trust_repo=True)  # or yolov5n - yolov5x6 or custom\n",
         "im = 'https://ultralytics.com/images/zidane.jpg'  # file, Path, PIL.Image, OpenCV, nparray, list\n",
         "results = model(im)  # inference\n",
         "results.print()  # or .show(), .save(), .crop(), .pandas(), etc."

diff --git a/tutorial.ipynb b/tutorial.ipynb
@@ -602,4 +602,4 @@
       "outputs": []
     }
   ]
-}
+}