#623·examples

mnist 示例包含导致错误结果的不良实践

作者: kirk86创建于 2019年8月27日更新于 2025年10月21日
标签good first issue

在 mnist 示例中,应该明确说明 `F.nll_loss` 的默认性质,即平均迭代批中的结果。另一个可能导致错误的地方是 `len(train_loader.dataset)`,尽管它会返回 `train dataset` 中的样本数,如果有人使用 `SubsetRandomSampler` 将训练数据分成训练/验证集,那么再次调用 `len(train_loader.dataset)` 仍会返回训练集中的全部样本数(这是错误的),而不会反映正确的样本数,因为训练集现在包含 X% 少的样本,这些样本已作为验证集保留。

内容来源: pytorch/examples