torch.set_warn_always

torch.set_warn_always(b, /)

当此标志为 False（默认值）时，某些 PyTorch 警告可能每个进程只出现一次。这有助于避免过多的警告信息。将其设置为 True 会导致这些警告始终出现，这在调试时可能会有所帮助。

参数

b (bool) – 如果为 True，强制警告始终发出；如果为 False，则设置为默认行为