5.6.2 AlexNet最后卷积层的输出大小应该为256*6*6

ShusenTang / Dive-into-DL-PyTorch

本项目将《动手学深度学习》(Dive into Deep Learning)原书中的MXNet实现改为PyTorch实现。

http://tangshusen.me/Dive-into-DL-PyTorch

Apache License 2.0

18.41k stars 5.4k forks source link

Closed Tartisan closed 2 years ago

Tartisan commented 3 years ago

AlexNet最后卷积层的输出大小

AlexNet最后卷积层的输出应该是256 6 6吧， (13 - 3) / 2 + 1 = 6

我还没有训练，但是原文按照 256 5 5 来训练没问题吗？

版本信息 pytorch: 1.8.1 torchvision: 0.8.0 torchtext: ...