标签 > Expected input batch_size (1) to match target batch_size (4):先找模型里丢失的 batch 维
PyTorch 报错 Expected input batch_size (1) to match target batch_size (4)定位模型丢失 ba
这个报错的核心原因就是模型前向传播过程里 batch 维度被意外抹掉了,输入标签 target 的 batch 是 4,但是喂给 loss 函数的模型
2026-11-09 16:50:23
