From 5f6d037a80d1e8da8309370e4d47f7c58e6f6314 Mon Sep 17 00:00:00 2001 From: baiyfbupt Date: Fri, 10 Aug 2018 10:13:54 +0800 Subject: [PATCH] fix scripts --- fluid/image_classification/.run_ce.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fluid/image_classification/.run_ce.sh b/fluid/image_classification/.run_ce.sh index e54fcc8b..f06fdf76 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 -- GitLab