提交 1ff97432 编写于 作者: H Hui Zhang

export under cpu

上级 4c838379
...@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册