Skip to content

Commit 465734c

Browse files
committed
Fix mapping for resume_ckpt_path
1 parent be15732 commit 465734c

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

sleap_nn/config/trainer_config.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -326,12 +326,6 @@ def trainer_mapper(legacy_config: dict) -> TrainerConfig:
326326
"""
327327
legacy_config_optimization = legacy_config.get("optimization", {})
328328
legacy_config_outputs = legacy_config.get("outputs", {})
329-
resume_ckpt_path = legacy_config.get("model", {}).get("base_checkpoint", None)
330-
resume_ckpt_path = (
331-
(Path(resume_ckpt_path) / "best.ckpt").as_posix()
332-
if resume_ckpt_path is not None
333-
else None
334-
)
335329
run_name = legacy_config_outputs.get("run_name", None)
336330
run_name = run_name if run_name is not None else ""
337331
run_name_prefix = legacy_config_outputs.get("run_name_prefix", "")
@@ -423,7 +417,6 @@ def trainer_mapper(legacy_config: dict) -> TrainerConfig:
423417
Path(legacy_config_outputs.get("runs_folder", "."))
424418
).as_posix()
425419
trainer_cfg_args["run_name"] = run_name
426-
trainer_cfg_args["resume_ckpt_path"] = resume_ckpt_path
427420

428421
trainer_cfg_args["optimizer_name"] = re.sub(
429422
r"^[a-z]",

0 commit comments

Comments
 (0)