怎样解决这个错误·?UserWarning: The default behavior for interpolate/upsample with float scale_factor changed in 1.6.0 to align with other frameworks/libraries, and now uses scale_factor directly, instead of relying on the computed output size. If you wish to restore the old behavior, please set recompute_scale_factor=True. See the documentation of nn.Upsample for details. warnings.warn(
时间: 2023-06-15 08:03:51 浏览: 2291
Pytorch上下采样函数--interpolate用法
这个警告是由于 PyTorch1.6.0 版本中对 `nn.Upsample` 的默认行为改变导致的。如果你想恢复旧的行为,可以将 `recompute_scale_factor` 参数设置为 `True`,这样 PyTorch 将会在每次调用 `nn.Upsample` 时重新计算输出大小。具体的做法是在调用 `nn.Upsample` 时加上参数 `recompute_scale_factor=True`,例如:
```python
upsample = nn.Upsample(scale_factor=2, mode='nearest', recompute_scale_factor=True)
```
另外,你也可以通过设置环境变量 `PYTHONWARNINGS=ignore::UserWarning` 来忽略这个警告。
阅读全文