ValueError: Expected input batch_size (258) to match target batch_size (1).
时间: 2023-11-17 19:05:44 浏览: 141
查看xgb特征重要性输出全是nan,ValueError:’Booster.get_score() results in empty’ 的原因及解决方案
这个错误通常是由于输入数据和目标数据的batch_size不匹配引起的。请检查你的代码,确保输入数据和目标数据的batch_size相同。你可以使用torch.utils.data.DataLoader中的batch_size参数来设置它们的大小。如果你的代码已经正确设置了batch_size,那么请检查你的目标数据是否正确地被处理成了(batch_size, ...)的形状。
阅读全文