diff --git a/tests/prepare.sh b/tests/prepare.sh
index 4c2aa052a57923e3354847c07f2091bb3c0e5666..57e9a949d6bbb2beb8486f42e556e8fd94fe7365 100644
--- a/tests/prepare.sh
+++ b/tests/prepare.sh
@@ -42,6 +42,9 @@ if [ ${MODE} = "lite_train_infer" ] || [ ${MODE} = "whole_infer" ];then
     cd ILSVRC2012 
     mv train.txt train_list.txt
     mv val.txt val_list.txt
+    if [ ${MODE} = "lite_train_infer" ];then
+	cp -r train/* val/
+    fi
     cd ../../
 elif [ ${MODE} = "infer" ] || [ ${MODE} = "cpp_infer" ];then
     # download data