Update Training_and_inference_on_an_example_dataset.ipynb

murthylab · Dec 15, 2020 · 98fc109 · 98fc109
1 parent a772bab
commit 98fc109
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Training_and_inference_on_an_example_dataset.ipynb b/Training_and_inference_on_an_example_dataset.ipynb
@@ -122,7 +122,7 @@
         "colab": {}
       },
       "source": [
-        "!python3 -m pip install sleap"
+        "!pip install sleap==1.0.10"
       ],
       "execution_count": 0,
       "outputs": []
@@ -206,7 +206,7 @@
         "colab": {}
       },
       "source": [
-        "!sleap-train baseline.topdown_confmaps.json \"dataset/drosophila-melanogaster-courtship/courtship_labels.slp\" --run_name \"courtship.topdown_confmaps\" --video-paths \"dataset/drosophila-melanogaster-courtship/20190128_113421.mp4\""
+        "!sleap-train baseline_medium_rf.topdown.json \"dataset/drosophila-melanogaster-courtship/courtship_labels.slp\" --run_name \"courtship.topdown_confmaps\" --video-paths \"dataset/drosophila-melanogaster-courtship/20190128_113421.mp4\""
       ],
       "execution_count": 0,
       "outputs": []
@@ -414,4 +414,4 @@
       ]
     }
   ]
-}
+}