cp dir when it exits

4 years ago · 3c6eea077b
parent 1b84f21ccf
commit 3c6eea077b
1 changed files with 3 additions and 1 deletions
--- a/utils/tarball.sh
+++ b/utils/tarball.sh
@ -19,7 +19,9 @@ function clean() {
 trap clean EXIT
 # ckpt_prfix dir
-cp -r ${ckpt_prefix} ${output}
+if [ -d ${ckpt_prefix} ];then
    cp -r ${ckpt_prefix} ${output}
 fi
 # ckpt_prfix.{json,...}
 cp ${ckpt_prefix}.*  ${output}
 # model config, mean std, vocab