diff --git a/fluid/language_model/.run_ce.sh b/fluid/language_model/.run_ce.sh index c0cb8f5473a100ebf0e5c0697f9ab5e65e97560d..5ee2d8aa0582b2b8504f9ba645b6252aa75f23bf 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