PyTorch使用denseNet做图像classification中的小技巧

labels.data.view_as(pred) 让预测的维度和labels的维度一样,等于labels.data.view(pred.size())

image.png

你可能感兴趣的:(PyTorch使用denseNet做图像classification中的小技巧)