name | type | Input size |
Conv block1 | 2Conv + pool | 224 × 224 × 3 |
Conv block2 | 2Conv + pool | 112 × 112 × 128 |
Conv block3 | 3Conv + pool | 56 × 56 × 256 |
Conv block4 | 3Conv + pool | 28 × 28 × 512 |
Conv block5 | 3Conv + pool | 14 × 14 × 512 |
Flatten | Flatten | 7 × 7 × 512 |
linear | linear | 1 × 1 × 4096 |
softmax | softmax | 1 × 1 × 1000 |