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

 更新时间:2019年12月31日 18:01:01   作者:KevinUSTC  
今天小编就为大家分享一篇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()

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

以上这篇PyTorch实现更新部分网络,其他不更新就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • python 无损批量压缩图片(支持保留图片信息)的示例

    python 无损批量压缩图片(支持保留图片信息)的示例

    这篇文章主要介绍了python 无损批量压缩图片的示例,帮助大家更好的利用python处理图片,感兴趣的朋友可以了解下
    2020-09-09
  • opencv实践项目之图像拼接详细步骤

    opencv实践项目之图像拼接详细步骤

    OpenCV的应用领域非常广泛,包括图像拼接、图像降噪、产品质检、人机交互、人脸识别、动作识别、动作跟踪、无人驾驶等,下面这篇文章主要给大家介绍了关于opencv实践项目之图像拼接的相关资料,需要的朋友可以参考下
    2023-05-05
  • python读取图片任意范围区域

    python读取图片任意范围区域

    这篇文章主要为大家详细介绍了python读取图片任意范围区域,以一维数组形式返回,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-01-01
  • 基于Python3.6+splinter实现自动抢火车票

    基于Python3.6+splinter实现自动抢火车票

    这篇文章主要为大家详细介绍了基于Python3.6+splinter实现自动抢火车票,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-09-09
  • python实现音乐下载器

    python实现音乐下载器

    这篇文章主要为大家详细介绍了python实现音乐下载器,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-04-04
  • 使用Python & Flask 实现RESTful Web API的实例

    使用Python & Flask 实现RESTful Web API的实例

    下面小编就为大家带来一篇使用Python & Flask 实现RESTful Web API的实例。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-09-09
  • Python中模块与包有相同名字的处理方法

    Python中模块与包有相同名字的处理方法

    这篇文章主要给大家介绍了在Python中模块与包有相同名字的处理方法,文中介绍的非常详细,对大家具有一定的参考学习价值,需要的朋友们下面来一起看看吧。
    2017-05-05
  • 详解python读取和输出到txt

    详解python读取和输出到txt

    这篇文章主要介绍了python读取和输出到txt,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-03-03
  • 利用Python将数值型特征进行离散化操作的方法

    利用Python将数值型特征进行离散化操作的方法

    今天小编就为大家分享一篇利用Python将数值型特征进行离散化操作的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-11-11
  • 一篇教程教你学会Python进制转换(十进制转二进制、八进制、十六进制)

    一篇教程教你学会Python进制转换(十进制转二进制、八进制、十六进制)

    计算机中只有1和0,所以就导致很多时候需要进制转换,本文介绍了Python进制转换,十进制转二进制,十进制转八进制,十进制转十六进制,有兴趣的可以了解一下
    2021-05-05

最新评论