fix benchmark dataset (PaddlePaddle#464)

AnnaTrainingG · Nov 18, 2021 · c21b08f · c21b08f
1 parent 87537ad
commit c21b08f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/benchmark/README.md b/benchmark/README.md
@@ -19,6 +19,8 @@
 #### 备注
 BasicVSR模型因竞品torch模型只能测4卡，故这里也测4卡。
 
+因REDS数据集较大，避免每次下载时间较长，需要在Docker建立好后，将REDS数据集放到/workspace/data/目录一下。
+
 ### Docker 镜像
 
 - **镜像版本**: `registry.baidubce.com/paddlepaddle/paddle:2.1.2-gpu-cuda10.2-cudnn7`

diff --git a/benchmark/run_all.sh b/benchmark/run_all.sh
@@ -64,9 +64,7 @@ for model_mode in ${model_mode_list[@]}; do
       eval dataset_web='$'"${model_mode}_dataset_web"
       eval dataset='$'"${model_mode}_dataset"
       eval log_interval='$'"${model_mode}_log_interval"
-      if [ -n "$dataset" ]; then
-          cp -r ${dataset} data/
-      else
+      if [ -n "$dataset_web" ]; then
           wget ${dataset_web} -O data/${model_mode}.tar
           tar -vxf data/${model_mode}.tar -C data/
       fi