README.md 2.15 KB
Newer Older
liangjing's avatar
liangjing committed
1
2
3
4
5
6
7
8
9
10
11
12
# ResNet50(Residual Network with 50-layer)

## 模型介绍

ResNet50是一种用于图像识别的深度神经网络模型。

## 模型结构

ResNet50网络模型由一系列卷积层、池化层、全局平均池化层和全连接层组成。该模型的特殊之处在于它包含多个残差块,每个残差块由多个卷积层和跳跃连接组成。

在ResNet50中,输入图像通过一系列卷积层和池化层进行特征提取,然后通过多个残差块进行深度特征学习。每个残差块包含多个卷积层和跳跃连接,跳跃连接允许信息在残差块内和残差块之间的传递,从而解决了深度神经网络中出现的梯度消失问题。最后,全局平均池化层将特征映射到一个固定长度的向量,该向量再通过全连接层进行分类或回归等任务。

liangjing's avatar
liangjing committed
13
14
15
16
17
18
19
20
21
22
## 目标精度

75.90% classification

## MLPerf代码参考版本

版本:mlperf_benchmark 

原始代码位置:https://github.com/mlcommons/training/tree/master/image_classification/tensorflow2

liangjing's avatar
liangjing committed
23
24
25
26
27
28
29
30
31
32
## 数据集

需使用ImageNET数据集,下载地址:http://image-net.org/challenges/LSVRC/2012/2012-downloads (require an account)

具体处理方式可详见:https://github.com/mlcommons/training/tree/master/image_classification

## 训练

### 环境配置

liangjing's avatar
liangjing committed
33
34
35
36
单机8卡进行性能&&精度测试

### 环境配置

liangjing's avatar
liangjing committed
37
38
提供[光源](https://www.sourcefind.cn/#/service-details)拉取的训练的docker镜像:

liangjing's avatar
liangjing committed
39

liangjing's avatar
liangjing committed
40
    docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:mlperf-resnet50-mpirun-latest
liangjing's avatar
liangjing committed
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

python依赖安装:

    pip install -r requirement.txt

### 训练

训练命令:

    bash 8dcu_multi.sh >& output.log &

### 预训练模型

在本目录下我们提供了一个预训练模型如下:

    mlperf_resnet50
    ├── checkpoint
    ├── ckpt-0.data-00000-of-00001
    ├── ckpt-0.index
    ├── ckpt-500.data-00000-of-00001
    ├── ckpt-500.index

liangjing's avatar
liangjing committed
63
## 测试结果
liangjing's avatar
liangjing committed
64

liangjing's avatar
liangjing committed
65
采用上述输入数据,加速卡采用Z100L*8,可最终达到官方收敛要求;
liangjing's avatar
liangjing committed
66
67
68
69
70
71
72
73

## 历史版本

* https://developer.hpccube.com/codes/modelzoo/mlperf_resnet50

## 参考

* https://mlcommons.org/en/
liangjing's avatar
liangjing committed
74
* https://github.com/mlcommons