torch.set_warn_always

torch.set_warn_always(b, /)[源代码]

当此标志为 False(默认值)时,某些 PyTorch 警告在每个进程中只会出现一次,以减少多余的警告信息。将其设置为 True 则会使这些警告一直显示,有助于调试。

参数

b (bool) – 如果为 True,强制始终显示警告;如果为 False,则使用默认设置。

本页目录