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 |