weight tensor should be defined either for all 3 classes or no classes but got weight tensor of shape: [1 x 2] at C:/cb/pytorch_1000000000000/work/aten/src\THCUNN/generic/ClassNLLCriterion.cu:44
时间: 2024-01-24 17:02:34 浏览: 351
pytorch 改变tensor尺寸的实现
这个错误通常是由于使用 `weight` 参数时出现的维度不匹配导致的。`weight` 参数应该是一个形状为 `[num_classes]` 的张量,其中 `num_classes` 是分类问题的类别数。在你的情况下,你提供的 `weight` 参数形状是 `[1, 2]`,这意味着你正在尝试为两个类别中的一个分配不同的权重。你需要将 `weight` 参数的形状更改为 `[2]`,并确保每个类别都有一个相应的权重值。
阅读全文