AttributeError: module 'torch.nn' has no attribute 'sigmoid_cross_entropy_with_logits'
时间: 2023-08-06 11:07:15 浏览: 298
AttributeError: module 'tensorflow.compat.v1' has no attribute '
这个错误是由于您正在使用的torch版本较旧导致的。在较新的torch版本中,sigmoid_cross_entropy_with_logits已被移动到torch.nn.functional模块中。您可以尝试使用以下代码进行修复:
```python
import torch
import torch.nn.functional as F
# ...
loss = F.binary_cross_entropy_with_logits(logits, targets)
```
这将使用torch.nn.functional中的binary_cross_entropy_with_logits函数来计算二分类交叉熵损失。请确保您的torch版本大于等于1.6.0,以使用此函数。
阅读全文