提交 4e94debf 编写于 作者: H Hui Zhang

fix bugs

上级 88d82b4a
...@@ -13,7 +13,7 @@ ckpt_path_prefix=$2 ...@@ -13,7 +13,7 @@ ckpt_path_prefix=$2
jit_model_export_path=$3 jit_model_export_path=$3
device=gpu device=gpu
if [ ngpu != 0 ];then if [ ngpu == 0 ];then
device=cpu device=cpu
fi fi
......
...@@ -9,7 +9,7 @@ ngpu=$(echo $CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}') ...@@ -9,7 +9,7 @@ ngpu=$(echo $CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
echo "using $ngpu gpus..." echo "using $ngpu gpus..."
device=gpu device=gpu
if [ ngpu != 0 ];then if [ ngpu == 0 ];then
device=cpu device=cpu
fi fi
config_path=$1 config_path=$1
......
...@@ -12,9 +12,10 @@ config_path=$1 ...@@ -12,9 +12,10 @@ config_path=$1
ckpt_name=$2 ckpt_name=$2
device=gpu device=gpu
if [ ngpu != 0 ];then if [ ngpu == 0 ];then
device=cpu device=cpu
fi fi
echo "using ${device}..."
mkdir -p exp mkdir -p exp
......
...@@ -13,7 +13,7 @@ ckpt_path_prefix=$2 ...@@ -13,7 +13,7 @@ ckpt_path_prefix=$2
jit_model_export_path=$3 jit_model_export_path=$3
device=gpu device=gpu
if [ ngpu != 0 ];then if [ ngpu == 0 ];then
device=cpu device=cpu
fi fi
......
...@@ -9,7 +9,7 @@ ngpu=$(echo $CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}') ...@@ -9,7 +9,7 @@ ngpu=$(echo $CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
echo "using $ngpu gpus..." echo "using $ngpu gpus..."
device=gpu device=gpu
if [ ngpu != 0 ];then if [ ngpu == 0 ];then
device=cpu device=cpu
fi fi
config_path=$1 config_path=$1
......
...@@ -12,7 +12,7 @@ config_path=$1 ...@@ -12,7 +12,7 @@ config_path=$1
ckpt_name=$2 ckpt_name=$2
device=gpu device=gpu
if [ ngpu != 0 ];then if [ ngpu == 0 ];then
device=cpu device=cpu
fi fi
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册