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

fix bugs

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