未验证 提交 c21b08f5 编写于 作者: W wangna11BD 提交者: GitHub

fix benchmark dataset (#464)

上级 87537ad9
......@@ -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`
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册