Update lerobot/scripts/train.py

Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
Alexander Soare 2024-05-21 17:45:41 +01:00 committed by GitHub
parent 675476a9ef
commit 452d15da6c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -231,7 +231,7 @@ def train(cfg: DictConfig, out_dir: str | None = None, job_name: str | None = No
if cfg.resume:
if not logger.last_checkpoint_dir.exists():
raise RuntimeError(
f"You have set resume=True, but there is no model checpoint in {logger.last_checkpoint_dir}."
f"You have set resume=True, but there is no model checkpoint in {logger.last_checkpoint_dir}."
)
else:
checkpoint_cfg_path = str(logger.last_pretrained_model_dir / "config.yaml")