From 7d625f31dcbdc1aa0db1791c1a956ed0aef87110 Mon Sep 17 00:00:00 2001 From: guochaorong <32069604+guochaorong@users.noreply.github.com> Date: Thu, 9 Aug 2018 08:06:07 +0800 Subject: [PATCH] Update .run_ce.sh --- fluid/language_model/.run_ce.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fluid/language_model/.run_ce.sh b/fluid/language_model/.run_ce.sh index c0cb8f54..5ee2d8aa 100644 --- a/fluid/language_model/.run_ce.sh +++ b/fluid/language_model/.run_ce.sh @@ -6,9 +6,9 @@ export OMP_NUM_THREADS=1 cudaid=${language_model:=0} # use 0-th card as default export CUDA_VISIBLE_DEVICES=$cudaid -FLAGS_benchmark=true python train.py | python _ce.py +FLAGS_benchmark=true python train.py --enable_ce | python _ce.py cudaid=${language_model_m:=0,1,2,3} # use 0,1,2,3 card as default export CUDA_VISIBLE_DEVICES=$cudaid -FLAGS_benchmark=true python train.py | python _ce.py +FLAGS_benchmark=true python train.py --enable_ce | python _ce.py -- GitLab