pull/578/head
Hui Zhang 4 years ago
parent 88d82b4ab8
commit 4e94debf69

@ -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

Loading…
Cancel
Save