关于 Resnet 输出 logits 形状的困惑

Question

我试图理解为什么我正在使用的 Resnet18 模型的输出 logits 的形状是 (27, 19)。我理解 19 的形状，即我设置模型预测的类数，但 27 的形状是我感到困惑的部分。我的批次大小为 64，所以我认为 logits 的形状是 (64, 19)，因为这将为批次中的每个图像提供 1 个预测向量...

Answer 1

事实证明，我正在查看我的 epoch 中最后一批的 logits，没有足够的图像来填满整个 64 批次大小，因此只剩下 27 张图像可以训练。

Answer 2

你明白了。

Torch 数据加载器这样做是因为方法 drop_last 默认为 False。如果将其设置为 True，则只会输出 logits shape (64, 19)

https://pytorch.org/docs/stable/data.html