现在位置: 首页 > PyTorch 教程 > 正文

PyTorch torch.set_warn_always 函数


Pytorch torch 参考手册 Pytorch torch 参考手册

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("警告已恢复默认行为")

输出结果为:

警告已设置为始终显示
警告已恢复默认行为

Pytorch torch 参考手册 Pytorch torch 参考手册