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算法应用实战之栈详解
Feb 04 Python
Python实现合并同一个文件夹下所有PDF文件的方法示例
Apr 28 Python
python3 对list中每个元素进行处理的方法
Jun 29 Python
Python Pywavelet 小波阈值实例
Jan 09 Python
TensorFlow车牌识别完整版代码(含车牌数据集)
Aug 05 Python
python实现统计代码行数的小工具
Sep 19 Python
安装2019Pycharm最新版本的教程详解
Oct 22 Python
利用Tensorflow的队列多线程读取数据方式
Feb 05 Python
JupyterNotebook 输出窗口的显示效果调整方法
Apr 13 Python
python使用布隆过滤器的实现示例
Aug 20 Python
Python使用paramiko连接远程服务器执行Shell命令的实现
Mar 04 Python
python flask开发的简单基金查询工具
Jun 02 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使用GIFEncoder类生成的GIF动态图片验证码
2014/07/01 PHP
既简单又安全的PHP验证码 附调用方法
2016/06/02 PHP
ucenter中词语过滤原理分析
2016/07/13 PHP
常用PHP数组排序函数归纳
2016/08/08 PHP
php 7新特性之类型申明详解
2017/06/06 PHP
php 读取文件夹下所有图片、文件的实例
2018/10/17 PHP
在laravel-admin中列表中禁止某行编辑、删除的方法
2019/10/03 PHP
laravel excel 上传文件保存到本地服务器功能
2019/11/14 PHP
JavaScript CSS修改学习第二章 样式
2010/02/19 Javascript
关于onScroll事件在IE6下每次滚动触发三次bug说明
2011/09/21 Javascript
用js实现输入提示(自动完成)的实例代码
2013/06/14 Javascript
toggle一个div显示或隐藏且可扩展成自定义下拉框
2013/09/12 Javascript
javascript中JSON对象与JSON字符串相互转换实例
2015/07/11 Javascript
扩展jquery easyui tree的搜索树节点方法(推荐)
2016/10/28 Javascript
js学习笔记之事件处理模型
2016/10/31 Javascript
Validform表单验证总结篇
2016/10/31 Javascript
js实现文字跑马灯效果
2017/02/23 Javascript
php 修改密码实现代码
2017/05/24 Javascript
VsCode插件整理(小结)
2017/09/14 Javascript
10个在JavaScript开发中常遇到的BUG
2017/12/18 Javascript
微信小程序云开发 生成带参小程序码流程
2019/05/18 Javascript
简单了解JavaScript中的执行上下文和堆栈
2019/06/24 Javascript
Python抓取百度查询结果的方法
2015/07/08 Python
Windows下安装python MySQLdb遇到的问题及解决方法
2017/03/16 Python
使用Python & Flask 实现RESTful Web API的实例
2017/09/19 Python
Python即时网络爬虫项目启动说明详解
2018/02/23 Python
Python基础之文件读取的讲解
2019/02/16 Python
翻转数列python实现,求前n项和,并能输出整个数列的案例
2020/05/03 Python
英国在线潜水商店:Simply Scuba
2019/03/25 全球购物
优秀员工表扬信
2014/01/17 职场文书
家长通知书家长评语
2014/04/17 职场文书
社区党员公开承诺书
2014/08/30 职场文书
幼儿园教师节演讲稿
2014/09/03 职场文书
初中信息技术教学计划
2015/01/22 职场文书
js实现上传图片到服务器
2021/04/11 Javascript
golang协程池模拟实现群发邮件功能
2021/05/02 Golang