由于排版打印中的失误,书中第96页表格遗漏了一列,请参考如下.下一版印刷中已经对此错误已经改正,对读者阅读过程中的不便我们深感抱歉.
张量形状 | 张量占用内存(KB) | 可训练参数 | 参数内存占用(KB) | |
---|---|---|---|---|
input | 224x224x3 | 224x224x3x4/1024 = 588 | 0 | 0 |
conv3-64 | 224x224x64 | 224x224x64x4/1024 = 12, 544 | (3x3x3)x64+64 = 1,792 | 1792x4/1024=7 |
conv3-64 | 224x224x64 | 12, 544 | (3x3x64)x64+64 = 36,928 | 36928x4/1024=144.25 |
maxpool | 112x112x64 | 3136 | 0 | 0 |
conv3-128 | 112x112x128 | 6272 | (3x3x64)x128+128 = 73,856 | 288.5 |
conv3-128 | 112x112x128 | 6272 | (3x3x128)x128+128 = 147,584 | 576.5 |
maxpool | 56x56x128 | 1568 | 0 | 0 |
conv3-256 | 56x56x256 | 3136 | (3x3x128)x256+256= 295,168 | 1153 |
conv3-256 | 56x56x256 | 3136 | (3x3x256)x256+256= 590,080 | 2,305 |
conv3-256 | 56x56x256 | 3136 | (3x3x256)x256+256= 590,080 | 2,305 |
maxpool | 28x28x256 | 784 | 0 | 0 |
conv3-512 | 28x28x512 | 1568 | (3x3x256)x512+512= 1,180,160 | 4,610 |
conv3-512 | 28x28x512 | 1568 | (3x3x512)x512+512=2,359,808 | 9218 |
conv3-512 | 28x28x512 | 1568 | (3x3x512)x512+512=2,359,808 | 9218 |
maxpool | 14x14x512 | 392 | 0 | 0 |
conv3-512 | 14x14x512 | 392 | (3x3x512)x512+512=2,359,808 | 9218 |
conv3-512 | 14x14x512 | 392 | (3x3x512)x512+512=2,359,808 | 9218 |
conv3-512 | 14x14x512 | 392 | (3x3x512)x512+512=2,359,808 | 9218 |
maxpool | 7x7x512 | 98 | 0 | 0 |
FC-4096 | 1x1x4096 | 16 | 7x7x512x4096+4096 = 102,764,544 | 401424 |
FC-4096 | 1x1x4096 | 16 | 4096x4096 +4096 = 16,781,312 | 65552 |
FC-1000 | 1x1x1000 | 4096x1000 + 1000 = 4,097,000 | 16004 |