Layer (type)

Output Shape

Param

input (InputLayer)

[3, 224, 224]

0

block1-conv1 (Conv2d)

[64, 224, 224]

1792

block1-conv2 (Conv2d)

[64, 224, 224]

36,928

block1-pool (MaxPool2d)

[64, 112, 112]

0

block2-conv1 (Conv2d)

[128, 112, 112]

73,856

block2-conv2 (Conv2d)

[128, 112, 112]

147,584

block2-pool (MaxPool2d)

[128, 56, 56]

0

block3-conv1 (Conv2d)

[256, 56, 56]

295,168

block3-conv2 (Conv2d)

[256, 56, 56]

590,080

block3-conv3 (Conv2d)

[256, 56, 56]

590,080

block3-pool (MaxPool2d)

[256, 28, 28]

0

block4-conv1 (Conv2d)

[512, 28, 28]

1,180,160

block4-conv2 (Conv2d)

[512, 28, 28]

2,359,808

block4-conv3 (Conv2d)

[512, 28, 28]

2,359,808

block4-pool (MaxPool2d)

[512, 14, 14]

0

block5-conv1 (Conv2d)

[512, 14, 14]

2,359,808

block5-conv2 (Conv2d)

[512, 14, 14]

2,359,808

block5-conv3 (Conv2d)

[512, 14, 14]

2,359,808

block5-pool (MaxPool2d)

[512, 7, 7]

0

FC1 (Linear)

[4096, 1, 1]

102,764,544

FC2 (Linear)

[4096, 1, 1]

16,781,312

FC3 (Linear)

[4096, 1, 1]

4,097,000