tensorflow 只恢复部分模型参数的实例


Posted in Python onJanuary 06, 2020

我就废话不多说了,直接上代码吧!

import tensorflow as tf

def model_1():
  with tf.variable_scope("var_a"):
    a = tf.Variable(initial_value=[1, 2, 3], name="a")

  vars = [var for var in tf.trainable_variables() if var.name.startswith("var_a")]
  print(len(vars))
  return vars

def model_2():

  vars1 = model_1()

  with tf.variable_scope("var_b"):
    a = tf.Variable(initial_value=[1, 2, 3], name="a")

  vars2 = [var for var in tf.trainable_variables() if var.name.startswith("var")]
  print(len(vars2))
  return vars1


def pretrain_model1():
  print("-------- model 1 ------")
  vars = model_1()

  with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    saver = tf.train.Saver()
    saver.save(sess, "./model.ckpt")

def train_model2():
  print("-------- model 2 ------")

  model1_vars = model_2()

  with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    saver = tf.train.Saver(var_list=model1_vars)
    saver.restore(sess, "./model.ckpt")
    vars = sess.run([model1_vars])
    for var in vars:
      print(var)

step = 2
if step == 1:
  pretrain_model1()
else:
  train_model2()

以上这篇tensorflow 只恢复部分模型参数的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
详解duck typing鸭子类型程序设计与Python的实现示例
Jun 03 Python
Python 中开发pattern的string模板(template) 实例详解
Apr 01 Python
python中matplotlib实现最小二乘法拟合的过程详解
Jul 11 Python
Python程序运行原理图文解析
Feb 10 Python
python3爬取淘宝信息代码分析
Feb 10 Python
对Python中数组的几种使用方法总结
Jun 28 Python
详解Python做一个名片管理系统
Mar 14 Python
python文档字符串(函数使用说明)使用详解
Jul 30 Python
Python如何实现动态数组
Nov 02 Python
Python+OpenCV检测灯光亮点的实现方法
Nov 02 Python
Python监听键盘和鼠标事件的示例代码
Nov 18 Python
python第三方网页解析器 lxml 扩展库与 xpath 的使用方法
Apr 06 Python
春节到了 教你使用python来抢票回家
Jan 06 #Python
Tensorflow 自定义loss的情况下初始化部分变量方式
Jan 06 #Python
在flask中使用python-dotenv+flask-cli自定义命令(推荐)
Jan 05 #Python
通过 Python 和 OpenCV 实现目标数量监控
Jan 05 #Python
python中count函数简单用法
Jan 05 #Python
Python猴子补丁知识点总结
Jan 05 #Python
Python调用Windows API函数编写录音机和音乐播放器功能
Jan 05 #Python
You might like
php对gzip文件或者字符串解压实例参考
2008/07/25 PHP
两千行代码的PHP学习笔记汇总
2014/10/05 PHP
php合并数组并保留键值的实现方法
2018/03/12 PHP
php 输出缓冲 Output Control用法实例详解
2020/03/03 PHP
用ASP将SQL搜索出来的内容导出为TXT的代码
2007/07/27 Javascript
jQuery 工具函数学习资料
2010/04/29 Javascript
jquery ajax提交表单数据的两种实现方法
2010/04/29 Javascript
javascript cookie操作类的实现代码小结附使用方法
2010/06/02 Javascript
JS实现在Repeater控件中创建可隐藏区域的代码
2010/09/16 Javascript
JS Range HTML文档/文字内容选中、库及应用介绍
2011/05/12 Javascript
js数值和和字符串进行转换时可以对不同进制进行操作
2014/03/05 Javascript
node.js中的url.format方法使用说明
2014/12/10 Javascript
jQuery统计上传文件大小的方法
2015/01/24 Javascript
详解javascript高级定时器
2015/12/31 Javascript
jQuery文本框得到与失去焦点动态改变样式效果
2016/09/08 Javascript
JS验证图片格式和大小并预览的简单实例
2016/10/11 Javascript
vue2.0结合Element实现select动态控制input禁用实例
2017/05/12 Javascript
通过命令行生成vue项目框架的方法
2017/07/12 Javascript
JavaScript中正则表达式使数字、中文或指定字符高亮显示
2017/10/31 Javascript
vue.js-div滚动条隐藏但有滚动效果的实现方法
2018/03/03 Javascript
vue项目每30秒刷新1次接口的实现方法
2018/12/04 Javascript
详解nvm管理多版本node踩坑
2019/07/26 Javascript
详解如何在JS代码中消灭for循环
2019/12/11 Javascript
Vue中通过属性绑定为元素绑定style行内样式的实例代码
2020/04/30 Javascript
JSON获取属性值方法代码实例
2020/06/30 Javascript
解决vue项目axios每次请求session不一致的问题
2020/10/24 Javascript
python交互式图形编程实例(二)
2017/11/17 Python
ubuntu安装mysql pycharm sublime
2018/02/20 Python
python 构造三维全零数组的方法
2018/11/12 Python
tensorflow入门:TFRecordDataset变长数据的batch读取详解
2020/01/20 Python
详解使用python3.7配置开发钉钉群自定义机器人(2020年新版攻略)
2020/04/01 Python
荷兰皇家航空公司中国官网:KLM中国
2017/12/13 全球购物
印度尼西亚最好的小工具在线商店:Erafone.com
2019/03/26 全球购物
Etam俄罗斯:法国女士内衣和家居服网上商店
2019/10/30 全球购物
给校长的建议书600字
2014/05/15 职场文书
幼儿园校园小喇叭广播稿
2014/10/17 职场文书