From 76448c34beb93b7c3b9a9fa5f806c61813319a36 Mon Sep 17 00:00:00 2001
From: kolinwei <331911734@qq.com>
Date: Thu, 3 Jan 2019 10:52:17 +0800
Subject: [PATCH] update image_classification .run_ce.sh (#1589)

---
 fluid/PaddleCV/image_classification/.run_ce.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/fluid/PaddleCV/image_classification/.run_ce.sh b/fluid/PaddleCV/image_classification/.run_ce.sh
index 9ba9a4c2..cc0d894a 100755
--- a/fluid/PaddleCV/image_classification/.run_ce.sh
+++ b/fluid/PaddleCV/image_classification/.run_ce.sh
@@ -7,6 +7,7 @@ cudaid=${object_detection_cudaid:=0}
 export CUDA_VISIBLE_DEVICES=$cudaid
 python train.py --batch_size=${BATCH_SIZE} --num_epochs=5 --enable_ce=True --lr_strategy=cosine_decay | python _ce.py
 
+BATCH_SIZE=224
 cudaid=${object_detection_cudaid_m:=0, 1, 2, 3}
 export CUDA_VISIBLE_DEVICES=$cudaid
 python train.py --batch_size=${BATCH_SIZE} --num_epochs=5 --enable_ce=True --lr_strategy=cosine_decay | python _ce.py
-- 
GitLab