标签：hook

2026-01-27andy阅读(3)评论(0)

在深度学习模型训练过程中，梯度爆炸（Gradient Explosion）是一个常见且致命的问题。它通常表现为损失值突然变为 NaN 或 Inf，导致训练中断或模型权重迅速发散。PyTorch 提供了强大的工具 register_hook，...