Fix resume argument in entrypoint

2026-02-22 04:47:50 -03:00
parent b21ba2ec40
commit e798d822f5
1 changed files with 1 additions and 1 deletions
@@ -7,7 +7,7 @@ CKPT=""

 if compgen -G "${CKPT_DIR}/version_*/checkpoints/*.ckpt" > /dev/null; then
  CKPT=$(ls -t ${CKPT_DIR}/version_*/checkpoints/*.ckpt | head -n1)
-  RESUME_ARG="--trainer.resume_from_checkpoint $CKPT"
+  RESUME_ARG="--ckpt_path $CKPT"

  echo "Resuming: $CKPT"
 fi