update snapshotindex after training

ttngu207 · Nov 15, 2023 · 736f4d7 · 736f4d7
1 parent ded35b3
commit 736f4d7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/element_deeplabcut/train.py b/element_deeplabcut/train.py
@@ -318,6 +318,13 @@ def make(self, key):
                 latest_snapshot = int(snapshot.stem[9:])
                 max_modified_time = modified_time
 
+        # update snapshotindex in the config
+        dlc_config["snapshotindex"] = latest_snapshot
+        edit_config(
+            dlc_cfg_filepath,
+            {"snapshotindex": latest_snapshot},
+        )
+
         self.insert1(
             {**key, "latest_snapshot": latest_snapshot, "config_template": dlc_config}
         )