Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
ou.cs
PaddleGAN
提交
6d626fa7
P
PaddleGAN
项目概览
ou.cs
/
PaddleGAN
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleGAN
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleGAN
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
6d626fa7
编写于
9月 29, 2021
作者:
W
wangna11BD
提交者:
GitHub
9月 29, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add BasicVSR Benchmark (#440)
* add Benchmark * add BasicVSR Benchmark
上级
9ecba77b
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
59 addition
and
4 deletion
+59
-4
benchmark/README.md
benchmark/README.md
+37
-0
benchmark/benchmark.yaml
benchmark/benchmark.yaml
+9
-1
benchmark/run_all.sh
benchmark/run_all.sh
+13
-3
未找到文件。
benchmark/README.md
0 → 100644
浏览文件 @
6d626fa7
# PaddGAN模型性能复现
## 目录
```
├── README.md # 说明文档
├── benchmark.yaml # 配置文件,设置测试模型及模型参数
├── run_all.sh # 执行入口,测试并获取所有生成对抗模型的训练性能
└── run_benchmark.sh # 执行实体,测试单个分割模型的训练性能
```
## 环境介绍
### 物理机环境
-
单机(单卡、8卡)
-
系统:CentOS release 7.5 (Final)
-
GPU:Tesla V100-SXM2-32GB
*
8
-
CPU:Intel(R) Xeon(R) Gold 6271C CPU @ 2.60GHz
*
80
-
CUDA、cudnn Version: cuda10.2-cudnn7
#### 备注
BasicVSR模型因竞品torch模型只能测4卡,故这里也测4卡。
### Docker 镜像
-
**镜像版本**
:
`registry.baidubce.com/paddlepaddle/paddle:2.1.2-gpu-cuda10.2-cudnn7`
-
**paddle 版本**
:
`2.1.2`
-
**CUDA 版本**
:
`10.2`
-
**cuDnn 版本**
:
`7`
## 测试步骤
```
bash
bash benchmark/run_all.sh
```
## 输出
执行完成后,在PaddleGAN目录会产出模型训练性能数据的文件,比如
`esrgan_mp_bs32_fp32_8`
等文件。
benchmark/benchmark.yaml
浏览文件 @
6d626fa7
...
...
@@ -23,9 +23,17 @@ esrgan:
log_interval
:
10
edvr
:
dataset
_web
:
https://paddlegan.bj.bcebos.com/datasets/REDS.tar
dataset
:
data/REDS
config
:
configs/edvr_m_wo_tsa.yaml
fp_item
:
fp32
bs_item
:
4
64
total_iters
:
300
log_interval
:
10
basicvsr
:
dataset
:
data/REDS
config
:
configs/basicvsr_reds.yaml
fp_item
:
fp32
bs_item
:
2
4
total_iters
:
300
log_interval
:
10
benchmark/run_all.sh
浏览文件 @
6d626fa7
...
...
@@ -32,9 +32,14 @@ for model_mode in ${model_mode_list[@]}; do
eval
total_iters
=
'$'
"
${
model_mode
}
_total_iters"
eval
epochs
=
'$'
"
${
model_mode
}
_epochs"
eval
dataset_web
=
'$'
"
${
model_mode
}
_dataset_web"
eval
dataset
=
'$'
"
${
model_mode
}
_dataset"
eval
log_interval
=
'$'
"
${
model_mode
}
_log_interval"
wget
${
dataset_web
}
-O
data/
${
model_mode
}
.tar
tar
-vxf
data/
${
model_mode
}
.tar
-C
data/
if
[
-n
"
$dataset
"
]
;
then
cp
-r
${
dataset
}
data/
else
wget
${
dataset_web
}
-O
data/
${
model_mode
}
.tar
tar
-vxf
data/
${
model_mode
}
.tar
-C
data/
fi
if
[
-n
"
$total_iters
"
]
;
then
mode
=
"total_iters"
max_iter
=
$total_iters
...
...
@@ -52,7 +57,12 @@ for model_mode in ${model_mode_list[@]}; do
sleep
60
echo
"index is speed, 8gpus, run_mode is multi_process, begin,
${
model_name
}
"
run_mode
=
mp
CUDA_VISIBLE_DEVICES
=
0,1,2,3,4,5,6,7 bash benchmark/run_benchmark.sh
${
run_mode
}
${
bs_item
}
${
fp_item
}
${
mode
}
${
max_iter
}
${
model_mode
}
${
config
}
${
log_interval
}
basicvsr_name
=
basicvsr
if
[
${
model_mode
}
=
${
basicvsr_name
}
]
;
then
CUDA_VISIBLE_DEVICES
=
0,1,2,3 bash benchmark/run_benchmark.sh
${
run_mode
}
${
bs_item
}
${
fp_item
}
${
mode
}
${
max_iter
}
${
model_mode
}
${
config
}
${
log_interval
}
else
CUDA_VISIBLE_DEVICES
=
0,1,2,3,4,5,6,7 bash benchmark/run_benchmark.sh
${
run_mode
}
${
bs_item
}
${
fp_item
}
${
mode
}
${
max_iter
}
${
model_mode
}
${
config
}
${
log_interval
}
fi
sleep
60
done
done
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录