torch.autograd.detect

import torch

# 正向传播时：开启自动求导的异常侦测
torch.autograd.set_detect_anomaly(True)

# 反向传播时：在求导时开启侦测
with torch.autograd.detect_anomaly():
    loss.backward()

【推广】免费学中医，健康全家人

原文地址：https://www.cnblogs.com/wjjcjj/p/14484530.html