benchmark_cn.md 563 字节
Newer Older
Q
qizhaoaoe 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
# 1. Benchmark

## 1.1 软硬件环境

* 单机 8卡 A100(40G)
* CUDA 11.2
* CUDNN 8.1

## 1.2 数据集
- 测试使用的数据集为ImageNet.

## 1.3 指标


| Model |DType | Phase | Dataset | gpu | img/sec | Top1 Acc | Official |
| --- | --- | --- | --- | --- | --- | --- | --- |
| Swin-B |FP16 O1|pretrain  |ImageNet2012  |A100*N1C8  |  2155| 0.83362 | 0.835 |
| Swin-B |FP16 O2|pretrain  | ImageNet2012 | A100*N1C8 | 3006 | 0.83223     | 0.835 |

# 2. 相关使用说明

https://github.com/PaddlePaddle/PLSC/blob/master/task/classification/swin/README.md