逐点卷积1 × 1 | 512 | 28 × 28 × 256 | 28 × 28 × 512 |
深度卷积3 × 3 | 512 | 28 × 28 × 512 | 14 × 14 × 512 |
逐点卷积1 × 1 | 512 | 14 × 14 × 512 | 14 × 14 × 512 |
注意机制模块 | - | 14 × 14 × 512 | 14 × 14 × 512 |
逐点卷积1 × 1 | 1024 | 14 × 14 × 512 | 14 × 14 × 1024 |
深度卷积3 × 3 | 1024 | 14 × 14 × 1024 | 7 × 7 × 1024 |
逐点卷积1 × 1 | 1024 | 7 × 7 × 1024 | 7 × 7 × 1024 |
全连接层 | - | 7 × 7 × 1024 | 1 × 1 × 1024 |
Softmax | - | 1 × 1 × 1024 | 10 |