torchvision tutorial (#1185)

torchvision tutorial cleanup --------- Co-authored-by: yardeny-sony <[email protected]>
sony · Aug 22, 2024 · 4b0884b · 4b0884b
1 parent bd3ceb5
commit 4b0884b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 296 deletions.
diff --git a/...ls/notebooks/imx500_notebooks/pytorch/pytorch_torchvision_classification_for_imx500.ipynb b/...ls/notebooks/imx500_notebooks/pytorch/pytorch_torchvision_classification_for_imx500.ipynb
diff --git a/...ebooks/imx500_notebooks/pytorch/pytorch_torchvision_classification_model_for_imx500.ipynb b/...ebooks/imx500_notebooks/pytorch/pytorch_torchvision_classification_model_for_imx500.ipynb
@@ -101,10 +101,9 @@
    "metadata": {},
    "cell_type": "code",
    "source": [
-    "# from torchvision.models import mobilenet_v2, MobileNet_V2_Weights\n",
-    "from torchvision.models import efficientnet_b0, EfficientNet_B0_Weights\n",
+    "from torchvision.models import mobilenet_v2, MobileNet_V2_Weights\n",
     "\n",
-    "model = efficientnet_b0(weights=EfficientNet_B0_Weights)\n",
+    "model = mobilenet_v2(weights=MobileNet_V2_Weights)\n",
     "model.eval()"
    ],
    "id": "ea84e114c819dde0",