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

2020/4/23 更新:

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

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

print(loss.item())即可。

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

 

 

以下是当时原文:

显然如果你已经定义好了

LossFunction=torch.nn.SoftMarginLoss()

loss=LossFunction(prediction,self.Y)

那么你确实可以用

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

这什么鬼啊,我要纯数值!

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

嗯,真香。

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

这次一定要记住(很简单!!!)

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

就可以了!

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐