PyTorch中hook函数的学习笔记

register_backward_hook()

作用是获取神经网络反向传播过程中,各个模块 输入端和输出端的梯度值 。

你可能感兴趣的:(PyTorch学习,《南溪的目标检测学习笔记》,PyTorch)