网络编程
位置:首页>> 网络编程>> Python编程>> PyTorch实现更新部分网络,其他不更新

PyTorch实现更新部分网络,其他不更新

作者:KevinUSTC  发布时间:2022-06-15 21:01:33 

标签:PyTorch,更新,网络

torch.Tensor.detach()的使用

detach()的官方说明如下:

Returns a new Tensor, detached from the current graph.
The result will never require gradient.

假设有模型A和模型B,我们需要将A的输出作为B的输入,但训练时我们只训练模型B. 那么可以这样做:


input_B = output_A.detach()

它可以使两个计算图的梯度传递断开,从而实现我们所需的功能。

来源:https://blog.csdn.net/jinxin521125/article/details/83621268

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com