diff --git a/fluid/image_classification/.run_ce.sh b/fluid/image_classification/.run_ce.sh index e54fcc8b61d4e01befe7fa845a3bc7a5e315dc94..f06fdf769024aef68ab5278fafb956f859228038 100755 --- a/fluid/image_classification/.run_ce.sh +++ b/fluid/image_classification/.run_ce.sh @@ -3,8 +3,8 @@ # This file is only used for continuous evaluation. cudaid=${object_detection_cudaid:=0} export CUDA_VISIBLE_DEVICES=$cudaid -python train.py --batch_size=64 --num_passes=10 --total_images=6149 --enable_ce=True | python _ce.py +python train.py --batch_size=64 --num_epochs=10 --total_images=6149 --enable_ce=True | python _ce.py cudaid=${object_detection_cudaid:=0, 1, 2, 3} export CUDA_VISIBLE_DEVICES=$cudaid -python train.py --batch_size=64 --num_passes=10 --total_images=6149 --enable_ce=True | python _ce.py +python train.py --batch_size=64 --num_epochs=10 --total_images=6149 --enable_ce=True | python _ce.py