pytorch取出loss function的值，纯数值

------------------------------------------------------------------------------------------------

2020/4/23 更新：

什么鬼啊。当时为什么要用这么麻烦。

直接loss.item()不就行了么。

print(loss.item())即可。

--------------------------------------------------------------------------------------------------------

 

 

以下是当时原文：

显然如果你已经定义好了

LossFunction=torch.nn.SoftMarginLoss()

loss=LossFunction(prediction,self.Y)

那么你确实可以用

print(loss)来看到这个值，但是很坑爹，只能用这种形式。

这什么鬼啊，我要纯数值！

之前有过一次这个疑问，而且解决了，当时心想我靠这么简单的问题下一次我就记住了。

嗯，真香。

我果然忘记怎么把这个loss的数值给取出来了！

这次一定要记住（很简单！！！）

print(loss.detach().numpy())

就可以了！