diff --git a/notebooks/007_train_project.ipynb b/notebooks/007_train_project.ipynb index 199eeb43..8b349d9d 100644 --- 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,