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的输出格式化和进制转换介绍,感兴趣的同学赶快来看一看吧,对你有帮助的话记得收藏一下
    2022-01-01
  • Python中模块(Module)和包(Package)的区别详解

    Python中模块(Module)和包(Package)的区别详解

    这篇文章主要介绍了Python中模块(Module)和包(Package)的区别详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-08-08
  • 正确理解Python中if __name__ == ''__main__''

    正确理解Python中if __name__ == ''__main__''

    今天小编就为大家分享一篇关于正确理解Python中if __name__ == '__main__' ,小编觉得内容挺不错的,现在分享给大家,具有很好的参考价值,需要的朋友一起跟随小编来看看吧
    2019-01-01
  • Python中OpenCV图像特征和harris角点检测

    Python中OpenCV图像特征和harris角点检测

    Harris角点检测算子是于1988年由CHris Harris & Mike Stephens提出来的。在具体展开之前,不得不提一下Moravec早在1981就提出来的Moravec角点检测算子。本文重点给大家介绍OpenCV图像特征harris角点检测知识,一起看看吧
    2021-09-09
  • Django ORM 查询管理器源码解析

    Django ORM 查询管理器源码解析

    这篇文章主要介绍了Django ORM 查询管理器源码解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-08-08
  • OpenCV-Python实现怀旧滤镜与连环画滤镜

    OpenCV-Python实现怀旧滤镜与连环画滤镜

    很多时候通过ps可以做很多效果,今天我们来介绍使用OpenCV-Python实现怀旧滤镜与连环画滤镜,具有一定的参考价值,感兴趣的可以了解一下
    2021-06-06
  • 关于Python使用turtle库画任意图的问题

    关于Python使用turtle库画任意图的问题

    这篇文章主要介绍了Python turtle库画任意图,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2022-04-04
  • Python数据类型及常用方法

    Python数据类型及常用方法

    这篇文章主要介绍了Python数据类型及常用方法 ,文章主要给大家分享Python的一些常用数据类型,具有一定的参考价值,需要的小伙伴可以参考一下
    2022-04-04
  • python pandas 解析(读取、写入)CSV 文件的操作方法

    python pandas 解析(读取、写入)CSV 文件的操作方法

    这篇文章主要介绍了python pandas 解析(读取、写入) CSV 文件,本文通过示例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2022-12-12
  • Python基于内置库pytesseract实现图片验证码识别功能

    Python基于内置库pytesseract实现图片验证码识别功能

    这篇文章主要介绍了Python基于内置库pytesseract实现图片验证码识别功能,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-02-02

最新评论