扩展率

输出张量尺寸

步长

模块数量

输入张量尺寸

模块名称

-

112 × 112 × 32

2

1

224 × 224 × 3

Conv2d 3 × 3

2

56 × 56 × 96

2

1

112 × 112 × 32

Sandglass Block

6

56 × 56 × 144

1

1

56 × 56 × 96

Sandglass Block

6

28 × 28 × 192

2

3

56 × 56 × 144

Sandglass Block

6

14 × 14 × 288

2

3

28 × 28 × 192

Sandglass Block

6

14 × 14 × 384

1

4

14 × 14 × 288

Sandglass Block

6

7 × 7 × 576

2

4

14 × 14 × 384

Sandglass Block

6

7 × 7 × 960

1

2

7 × 7 × 576

Sandglass Block

6

7 × 7 × 1280

1

1

7 × 7 × 960

Sandglass Block

-

1 × 1 × 1280

-

1

7 × 7 × 1280

Avgpool 7 × 7

-

k

-

1

1 × 1 × 1280

Conv2d 1 × 1