wlltmrt/piper-train
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix resume argument in entrypoint

Browse Source
This commit is contained in:
Wellington Marthas
2026-02-22 04:47:50 -03:00
parent b21ba2ec40
commit e798d822f5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
entrypoint.sh
+1 -1
View File
@@ -7,7 +7,7 @@ CKPT=""
if compgen -G "${CKPT_DIR}/version_*/checkpoints/*.ckpt" > /dev/null; then if compgen -G "${CKPT_DIR}/version_*/checkpoints/*.ckpt" > /dev/null; then
CKPT=$(ls -t ${CKPT_DIR}/version_*/checkpoints/*.ckpt | head -n1) 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" echo "Resuming: $CKPT"
fi fi