modify latest snapshot determination

datajoint · Nov 30, 2023 · 6278148 · 6278148
1 parent 474269e
commit 6278148
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/element_deeplabcut/train.py b/element_deeplabcut/train.py
@@ -326,15 +326,17 @@ def make(self, key):
         # Here, we mean most recently generated
 
         # `snapshotindex` refers to the file index of the snapshot generated
-        # The most recent snapshot index will be the number of snapshots
+        # The most recent snapshot index will be the number of snapshots - 1
+        # because it is an index
 
         # update snapshotindex in the config
-        dlc_config["snapshotindex"] = len(snapshots)
+        latest_snapshot = len(snapshots)-1
+        dlc_config["snapshotindex"] = latest_snapshot
         edit_config(
             dlc_cfg_filepath,
-            {"snapshotindex": len(snapshots)},
+            {"snapshotindex": latest_snapshot},
         )
 
         self.insert1(
-            {**key, "latest_snapshot": len(snapshots), "config_template": dlc_config}
+            {**key, "latest_snapshot": latest_snapshot, "config_template": dlc_config}
         )