ShusenTang / Dive-into-DL-PyTorch

本项目将《动手学深度学习》(Dive into Deep Learning)原书中的MXNet实现改为PyTorch实现。
http://tangshusen.me/Dive-into-DL-PyTorch
Apache License 2.0
18.25k stars 5.39k forks source link

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

Closed Tartisan closed 2 years ago

Tartisan commented 2 years ago

AlexNet最后卷积层的输出大小

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

image

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

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