【RuntimeError】the derivative for 'target' is not implemented

报错信息:
Traceback (most recent call last):  
  File "/users/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in __call__
    result = self.forward(*input, **kwargs)
  File "/users/anaconda3/lib/python3.6/site-packages/torch/nn/modules/loss.py", line 504, in forward
    return F.binary_cross_entropy(input, target, weight=self.weight, reduction=self.reduction)
  File "/users4/zsun/anaconda3/lib/python3.6/site-packages/torch/nn/functional.py", line 2027, in binary_cross_entropy
    input, target, weight, reduction_enum)
RuntimeError: the derivative for 'target' is not implemented

出错代码为:

torch.nn.BCELoss(preds, targets)

错误为:

loss函数的target,即第二项targets的导数不应该有梯度

查错误:

print(preds.requires_grad,targets.requires_grad);

发现结果为True,True

然而loss函数的preds.requires_grad=True,targets.requires_grad应该为False

改正:

加入一行targets = targets.detach()

NOTE:

targets.requires_grad = False不可以

原因是:

you can only change requires_grad flags of leaf variables.

If you want to use a computed variable in a subgraph that doesn't require differentiation please use detach()

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值