From 31f3ed7fde793699616d268bd228b11d105bdb57 Mon Sep 17 00:00:00 2001
From: dongshuilong <dongshuilong@baidu.com>
Date: Mon, 18 Oct 2021 10:18:43 +0000
Subject: [PATCH] fix autolog warmup bug for whole train lite_train_infer mode

---
 tests/prepare.sh | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/prepare.sh b/tests/prepare.sh
index 4c2aa052..57e9a949 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
-- 
GitLab