torch.autograd.set_detect_anomaly(True)

`torch.autograd.set_detect_anomaly(True)` 是一个 PyTorch 函数，用于在计算图中开启异常检测功能。如果在计算图中出现了异常（如 NaN 值或者梯度爆炸），PyTorch 会抛出异常并输出计算图中出错的操作节点，帮助用户快速定位问题所在。

需要注意的是，开启异常检测功能会对性能有一定影响，因此只在调试阶段使用。在模型训练和测试时最好关闭该功能。