Update ATSS algo name

openvinotoolkit · Aug 31, 2023 · 2082531 · 2082531
1 parent 176fd62
commit 2082531
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/notebooks/007_train_project.ipynb b/notebooks/007_train_project.ipynb
@@ -138,7 +138,9 @@
    "id": "71c6a907-eabe-4458-a667-4a78428a2213",
    "metadata": {},
    "source": [
-    "Let's go with the `ATSS` algorithm, which is a larger and more accurate model than the `SSD` one. Because of its size it is also slower, but let's say we care most about accuracy for now."
+    "Let's go with the `ATSS` algorithm, which is a larger and more accurate model than the `SSD` one. Because of its size it is also slower, but let's say we care most about accuracy for now.\n",
+    "\n",
+    "Note that later Geti versions refer to this algorithm under a slightly different name, so we have to wrap this in a `try / except` statement."
    ]
   },
   {
@@ -148,7 +150,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "algorithm = available_algorithms.get_by_name(name=\"ATSS\")"
+    "try:\n",
+    "    algorithm = available_algorithms.get_by_name(name=\"ATSS\")\n",
+    "except ValueError:\n",
+    "    algorithm = available_algorithms.get_by_name(name=\"MobileNetV2-ATSS\")"
    ]
   },
   {
@@ -292,7 +297,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.16"
+   "version": "3.8.17"
   }
  },
  "nbformat": 4,