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 相关文章推荐
Python爬虫爬验证码实现功能详解
Apr 14 Python
python实现下载pop3邮件保存到本地
Jun 19 Python
python最小生成树kruskal与prim算法详解
Jan 17 Python
Python使用dict.fromkeys()快速生成一个字典示例
Apr 24 Python
使用Python-OpenCV向图片添加噪声的实现(高斯噪声、椒盐噪声)
May 28 Python
Python将文字转成语音并读出来的实例详解
Jul 15 Python
Python 网络编程之TCP客户端/服务端功能示例【基于socket套接字】
Oct 12 Python
浅析python内置模块collections
Nov 15 Python
Python笔记之观察者模式
Nov 20 Python
python如何将两张图片生成为全景图片
Mar 05 Python
Ubuntu中配置TensorFlow使用环境的方法
Apr 21 Python
浅析Python实现DFA算法
Jun 26 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数据库连接类~~做成一个分页类!
2006/11/25 PHP
PHP.ini中配置屏蔽错误信息显示和保存错误日志的例子
2014/05/12 PHP
php导入大量数据到mysql性能优化技巧
2014/12/29 PHP
PHP实现的方程求解示例分析
2016/11/11 PHP
Yii框架扩展CGridView增加导出CSV功能的方法
2017/05/24 PHP
php压缩文件夹最新版
2018/07/18 PHP
学习ExtJS Panel常用方法
2009/10/07 Javascript
关于html+ashx开发中几个问题的解决方法
2011/07/18 Javascript
js购物车实现思路及代码(个人感觉不错)
2013/12/23 Javascript
php+js实现倒计时功能
2014/06/02 Javascript
jquery动态调整div大小使其宽度始终为浏览器宽度
2014/06/06 Javascript
Jquery 分页插件之Jquery Pagination
2015/08/25 Javascript
AngularJS 使用 UI Router 实现表单向导
2016/01/29 Javascript
Angular.js中用ng-repeat-start实现自定义显示
2016/10/18 Javascript
简单实现js悬浮导航效果
2017/02/05 Javascript
js实现彩色条纹滚动条效果
2017/03/15 Javascript
vue.js中过滤器的使用教程
2017/06/08 Javascript
ReactJS实现表单的单选多选和反选的示例
2017/10/13 Javascript
webpack 插件html-webpack-plugin的具体使用
2018/04/09 Javascript
js设置默认时间跨度过程详解
2019/07/17 Javascript
JS面向对象编程实现的拖拽功能案例详解
2020/03/03 Javascript
Python的Bottle框架中获取制定cookie的教程
2015/04/24 Python
编写Python CGI脚本的教程
2015/06/29 Python
一键搞定python连接mysql驱动有关问题(windows版本)
2016/04/23 Python
详解python websocket获取实时数据的几种常见链接方式
2019/07/01 Python
基于Python+Appium实现京东双十一自动领金币功能
2019/10/31 Python
HTML5样式控制示例代码
2013/11/27 HTML / CSS
Html5导航栏吸顶方案原理与对比实现
2020/06/10 HTML / CSS
Giglio德国网上精品店:奢侈品服装和配件
2016/09/23 全球购物
美国电子产品折扣网站:Daily Steals
2017/05/20 全球购物
eDreams加拿大:廉价航班、酒店和度假
2019/03/29 全球购物
活动邀请函范文
2014/01/19 职场文书
卖车协议书
2014/04/21 职场文书
国际金融专业自荐信
2014/07/05 职场文书
2014红色之旅心得体会
2014/10/07 职场文书
《我和小伙伴》教学反思
2016/02/20 职场文书