fix autolog warmup bug for whole train lite_train_infer mode

2021-10-18 10:18:43 +00:00 · 2021-10-18 10:18:43 +00:00 · 31f3ed7fde
parent 98bcc91597
commit 31f3ed7fde
1 changed files with 3 additions and 0 deletions
--- 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