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