1. 首页
  2. 编程语言
  3. Python
  4. PyTorch实现更新部分网络,其他不更新

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

上传者: 2021-05-04 11:56:32上传 PDF文件 39.99 KB 热度 19次

detach()的官方说明如下:Returns a new Tensor, detached from the current graph.假设有模型A和模型B,我们需要将A的输出作为B的输入,但训练时我们只训练模型B. 那么可以这样做:它可以使两个计算图的梯度传递断开,从而实现我们所需的功能。

用户评论