torch.autograd.detect_anomaly()

本文转载自查看原文 2021-03-05 09:57 2891 深度学习~pytorch

import torch

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

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

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。

猜您在找 PyTorch笔记--torch.autograd.grad() PyTorch 源码解读之 torch.autograd PYTORCH: 60分钟 | TORCH.AUTOGRAD pytorch-x,norm(),torch.autograd.grad函数 Pytorch中torch.autograd ---backward函数的使用方法详细解析，具体例子分析 Detect to Track and Track to Detect Anomaly Detection torch Autograd: 自动求导 PyTorch教程之Autograd