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 回调函数和回调方法的实现分析
Mar 23 Python
Python语言实现将图片转化为html页面
Dec 06 Python
Python通过OpenCV的findContours获取轮廓并切割实例
Jan 05 Python
python 接口测试response返回数据对比的方法
Feb 11 Python
python装饰器常见使用方法分析
Jun 26 Python
Python爬虫动态ip代理防止被封的方法
Jul 07 Python
Python PyQt5整理介绍
Apr 01 Python
python:解析requests返回的response(json格式)说明
Apr 30 Python
使用Numpy对特征中的异常值进行替换及条件替换方式
Jun 08 Python
如何使用Pytorch搭建模型
Oct 26 Python
Python 调用 ES、Solr、Phoenix的示例代码
Nov 23 Python
Python页面加载的等待方式总结
Feb 28 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
POSIX 风格和兼容 Perl 风格两种正则表达式主要函数的类比(preg_match, preg_replace, ereg, ereg_replace)
2010/10/12 PHP
PHP学习之数组的定义和填充
2011/04/17 PHP
php实现refresh刷新页面批量导入数据的方法
2014/12/23 PHP
php获取汉字拼音首字母的方法
2015/10/21 PHP
php验证邮箱和ip地址最简单方法汇总
2015/10/30 PHP
PHP与SQL语句写一句话木马总结
2019/10/11 PHP
学习并汇集javascript匿名函数
2010/11/25 Javascript
javascript 学习笔记(八)javascript对象
2011/04/12 Javascript
JavaScript面向对象(极简主义法minimalist approach)
2012/07/17 Javascript
jquery进行数组遍历如何跳出当前的each循环
2014/06/05 Javascript
jQuery遍历节点树方法分析
2016/09/08 Javascript
Bootstrap源码解读媒体对象、列表组和面板(10)
2016/12/26 Javascript
nodejs6下使用koa2框架实例
2017/05/18 NodeJs
nodejs log4js 使用详解
2019/05/31 NodeJs
Vue中图片Src使用变量的方法
2019/10/30 Javascript
解决idea开发遇到javascript动态添加html元素时中文乱码的问题
2020/09/29 Javascript
Python图像灰度变换及图像数组操作
2016/01/27 Python
python实现报表自动化详解
2017/11/16 Python
Python机器学习之K-Means聚类实现详解
2018/02/22 Python
离线状态下在jupyter notebook中使用plotly实例
2020/04/24 Python
在HTML5 canvas里用卷积核进行图像处理的方法
2018/05/02 HTML / CSS
HTML5 Canvas——用路径描画线条实例介绍
2013/06/09 HTML / CSS
美体小铺加拿大官方网站:The Body Shop加拿大
2016/10/30 全球购物
世界上第一个水枕头:Mediflow
2018/12/06 全球购物
俄罗斯达美乐比萨外送服务:Domino’s Pizza
2020/12/18 全球购物
学生打架检讨书1000字
2014/01/16 职场文书
旅游业大学生创业计划书
2014/01/31 职场文书
带病坚持工作事迹
2014/05/03 职场文书
企业法人代表任命书
2014/06/06 职场文书
百年孤独读书笔记
2015/06/29 职场文书
教师节校长致辞
2015/07/31 职场文书
2016应届毕业生实习心得体会
2015/10/09 职场文书
创业计划书之蛋糕店
2019/08/29 职场文书
二年级作文之动物作文
2019/11/13 职场文书
在 SQL 语句中处理 NULL 值的方法
2021/06/07 SQL Server