From 3c6eea077b2b077b9c3f5cc7baf339c545053d35 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Tue, 22 Jun 2021 07:27:43 +0000
Subject: [PATCH] cp dir when it exits

---
 utils/tarball.sh | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/utils/tarball.sh b/utils/tarball.sh
index a4611c75..5f7c21a3 100755
--- 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
-- 
GitLab