tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式

 更新时间:2020年01月24日 10:39:23   作者:曲草  
今天小编就为大家分享一篇tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

已经有了一个预训练的模型,我需要从其中取出某一层,把该层的weights和biases赋值到新的网络结构中,可以使用tensorflow中的pywrap_tensorflow(用来读取预训练模型的参数值)结合Session.assign()进行操作。

这种需求即预训练模型可能为单分支网络,当前网络为多分支,我需要把单分支A复用到到多个分支去(B,C,D)。

先导入对应的工具包

from tensorflow.python import pywrap_tensorflow

接下来的操作在一个tf.Session中进行

reader = pywrap_tensorflow.NewCheckpointReader(pre_train_model_path)

# 获取当前图可训练变量
trainable_variables = tf.trainable_variables()
# 需要赋值的当前网络层变量,这里只是随便起的名字。
restore_v_target_name = "fc_target"
# 需要的预训练模型中的某层的名字
restore_v_source_name = "fc_source"
for v in trainable_variables:
  if restore_v_target_name == v.name:
   # 回复weights和biases
    sess.run(
      tf.assign(v, reader.get_tensor(restore_v_source_name + "/weights"))) if "weights" in v.name else sess.run(
      tf.assign(v, reader.get_tensor(restore_v_source_name + "/biases")))

以上这篇tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • Python绘制地理图表可视化神器pyecharts

    Python绘制地理图表可视化神器pyecharts

    这篇文章主要介绍了Python绘制地理图表可视化神器pyecharts,文章围绕主题展开详细的内容介绍,具有一定的参考价值,需要的小伙伴可以参考一下
    2022-07-07
  • 如何用python获取EXCEL文件内容并保存到DBC

    如何用python获取EXCEL文件内容并保存到DBC

    很多时候,使用python进行数据分析的第一步就是读取excel文件,下面这篇文章主要给大家介绍了关于如何用python获取EXCEL文件内容并保存到DBC的相关资料,需要的朋友可以参考
    2023-12-12
  • Python何时应该使用Lambda函数

    Python何时应该使用Lambda函数

    这篇文章主要介绍了Python何时应该使用Lambda函数,Python 中定义函数有两种方法,一种是用常规方式 def 定义,函数要指定名字,第二种是用 lambda 定义,不需要指定名字,称为 Lambda 函数,需要的朋友可以参考下
    2019-07-07
  • 如何分离django中的媒体、静态文件和网页

    如何分离django中的媒体、静态文件和网页

    这篇文章主要介绍了如何分离django中的媒体、静态文件和网页,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-11-11
  • Python 多线程不加锁分块读取文件的方法

    Python 多线程不加锁分块读取文件的方法

    今天小编就为大家分享一篇Python 多线程不加锁分块读取文件的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-12-12
  • pytest fixtures装饰器的使用和如何控制用例的执行顺序

    pytest fixtures装饰器的使用和如何控制用例的执行顺序

    这篇文章主要介绍了pytest fixtures装饰器的使用和如何控制用例的执行顺序,帮助大家更好的理解和使用pytest测试框架,感兴趣的朋友可以了解下
    2021-01-01
  • Python中文分词工具之结巴分词用法实例总结【经典案例】

    Python中文分词工具之结巴分词用法实例总结【经典案例】

    这篇文章主要介绍了Python中文分词工具之结巴分词用法,结合实例形式总结分析了Python针对中文文件的读取与分词操作过程中遇到的问题与解决方法,需要的朋友可以参考下
    2017-04-04
  • PyQt5 QLineEdit输入的子网字符串校验QRegExp实现

    PyQt5 QLineEdit输入的子网字符串校验QRegExp实现

    这篇文章主要介绍了PyQt5 QLineEdit输入的子网字符串校验QRegExp实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2021-04-04
  • python实现AES算法及AES-CFB8加解密源码

    python实现AES算法及AES-CFB8加解密源码

    这篇文章主要为大家介绍了python实现AES算法及AES-CFB8加解密的源码示例,有需要朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步
    2022-02-02
  • 解决Python plt.savefig 保存图片时一片空白的问题

    解决Python plt.savefig 保存图片时一片空白的问题

    今天小编就为大家分享一篇解决Python plt.savefig 保存图片时一片空白的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-01-01

最新评论