训练新的数据集发生报错

ValueError: Using a target size (torch.Size([64, 24])) that is different to the input size (torch.Size([64, 17])) is deprecated. Please ensure they have the same size.

训练新的数据集发生报错_第1张图片

 原因是自己的标签只有17种,但是由于标注的时候误点了   变成了24种,清除掉脏数据就行了。

你可能感兴趣的:(python,pytorch,开发语言)