pytorch float object has no attribute backward

星月夜语

于 2020-06-08 21:38:34 发布

阅读量1.2w

点赞数 5

分类专栏： python 深度学习基础知识文章标签： pytorch myloss

本文链接：https://blog.youkuaiyun.com/ljh618625/article/details/106629273

版权

40 篇文章 ¥49.90 ¥99.00

订阅专栏

34 篇文章

订阅专栏

在使用PyTorch自定义loss函数时，遇到在反向传播过程中报错：'float' object has no attribute 'backward'。错误原因是loss函数返回了整型值而非浮点型。尽管实验中观察到梯度下降，但在某一时刻该错误打断了训练。解决方案可参考相关博客文章进行排查。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、问题描述

编写自己的loss 函数时， loss.backward() 在反向传播一会后，就报错：

'float' object has no attribute 'backward'

二、原因：

报错的原因是output，也就是损失函数这里输出了int值。但是在实验过程中，梯度确实是下下降了。只是总是在下降过程中出现了这种报错。

三、解决办法：

def  my loss(input):

  loss = np.sum(input)/len(input)
  return   (torch.tensor(0.0, requires_grad=True) if loss ==0  else loss)