PyTorch 报错 Expected input batch_size (1) to match target batch_size (4)定位模型丢失 ba

这个报错的核心原因就是模型前向传播过程里 batch 维度被意外抹掉了,输入标签 target 的 batch 是 4,但是喂给 loss 函数的模型