torch.nn.BCELoss(size_average=True)在新版本怎么改
时间: 2024-02-11 22:08:42 浏览: 136
Pytorch中torch.nn的损失函数
5星 · 资源好评率100%
在 PyTorch 1.0.0 版本中,`size_average` 参数已经被弃用,并且使用 `reduction` 参数来代替。如果你想在 PyTorch 的新版本中使用 `BCELoss`,你可以将 `size_average=True` 改为 `reduction='mean'`,代码如下:
```
loss = torch.nn.BCELoss(reduction='mean')
```
如果你想保持与旧版本的兼容性,可以使用 `size_average` 参数的替代参数 `reduce`,并将其设置为 `True`,代码如下:
```
loss = torch.nn.BCELoss(size_average=True, reduce=True)
```
不过,这种方式已经被弃用,不推荐使用。建议使用 `reduction` 参数来控制缩减方式。
阅读全文