PyTorch torch.set_warn_always 函数
torch.set_warn_always 是 PyTorch 中用于设置是否始终显示警告的函数。当启用时,PyTorch 会始终显示弃用警告,即使这些警告已经被忽略过。
函数定义
torch.set_warn_always(warn)
参数说明
warn: 布尔值,True 表示始终显示警告,False 表示默认行为
使用示例
实例
import torch
# 设置始终显示警告
torch.set_warn_always(True)
print("警告已设置为始终显示")
# 恢复默认行为
torch.set_warn_always(False)
print("警告已恢复默认行为")
# 设置始终显示警告
torch.set_warn_always(True)
print("警告已设置为始终显示")
# 恢复默认行为
torch.set_warn_always(False)
print("警告已恢复默认行为")
输出结果为:
警告已设置为始终显示 警告已恢复默认行为

Pytorch torch 参考手册