Tensorflow 合并通道及加载子模型的方法
作者：pl_Dong 发布时间：2023-08-10 05:36:54
标签：Tensorflow,通道,子模型
最近在使用Tensorflow 实现DNN网络时，遇到一些问题。目前网上关于Tensorflow的资料还比较少，现把问题和解决方法写出来，仅供参考。
（1）将两个子模型的输出合并到一个通道，例如同时连接到一个全连接层如图
合并方法为 tf.concat()函数。此函数需要两个个参数 concat（0或1，[合并节点1，合并节点2] ）。0 或 1 代表节点合并的方式：0 代表合并后列相同，行增加；1 代表合并后行相同，列增加。
上图所示合并方法为： X_20 = tf.concat(1, [X_top, X_down]);
（2）加载预训练好模型的部分参数。例如训练完成了一个五层网络，现在需要训练好一个七层网络，使用已训练好的五层网络参数初始化七层网络的前五层。
首先，五层网络保存模型时的参数变量名要和七层网络的需要初始化的参数变量名保持一致。
然后，定义加载指定变量名的 tf.train.Saver()。
例如：
Saver在restore模型时就会只加载 W_fc1_I, W_fc2_I, b_fc1_I, b_fc2_I 这四个参数，并且初始化当前模型中变量名相同的变量。需要注意的是，其他参数也是需要初始化的，一种避免遗漏初始化变量的方法是首先使用 initialize_all_variables() 函数对所有参数进行初始化之后再调用 restore 函数初始化需要的部分参数。
此方法只是其中一种解决方法，仅供参考。
来源：https://blog.csdn.net/Perry_PKU/article/details/52325774
0
投稿猜你喜欢
- forms组件django框架提供了一个Form类，来进行web开发中的表单提交数据的处理工作。导入相关模块from django impo
饼图中的series有个avoidLabelOverlap属性，avoidLabelOverlap:是否启用防止标签重叠策略，默认开启，在标- 关于这个问题的解答，网络上有成千上万个版本，当然，出现问题的诱因不同，解决的途径也不同，所以我不排除其他解决此类问题的办法，但是这些办法都没
- 本文主要研究的是selenium python浏览器多窗口处理的相关内容，分享了操作实例代码，具体如下：#!/usr/bin/python#
- 简述1.pythonpython作为一门解释型脚本语言，它有三种发布方式：文件 : 源码文件，运行需要使用者安装Python环境并且安装依赖
- 本文实例讲述了Python栈的实现方法。分享给大家供大家参考，具体如下：Python实现栈栈的数组实现：利用python列表方法代码如下：#
- 在python中Template可以将字符串的格式固定下来，重复利用。 同一套测试框架为了可以复用，所以我们可以将用例部分做参数化，然后运用
- 整数的阶乘（英语：factorial）是所有小于及等于该数的正整数的积，0的阶乘为1。即：n!=1×2×3×...×n。首先导入math模块
- 本文详细介绍了asp中如何使用sql语句删除数据库中的记录，初学asp者来看看！1，首先要明确删除哪条记录无非还就是SQL语句了，比如对应到
- 在学习OpenCV或者其他关于Python技术的时候，我们通常需要准备不同的Python环境，我选择了Anaconda作为我的Python环
- 最近的一个页面中碰到的，本来想用 border 来模拟设计图的虚线效果，但是很明显 border 效果不如设计图来的好看。顺便研究了下 da
- 如下所示：import numpy as npx = [1,2] #横坐标y = [3,4] #第一个纵坐标y1 =
- 一、查询某个字段重复 select * from User u whe
- 首先让我们看下 YUI 是如何处理的：var toObject = function(a) { var o = {
- 我就废话不多说了，大家还是直接看代码吧~/** 二分幂法 求x^n */// 求整数幂package mainimport (
- 一、操作步骤导入：import flask,json实例化：api = flask.Flask(name)定义接口访问路径及访问方式：@ap
- 1. resize函数说明 OpenCV提供了resize函数来改变图像的大小，函数原型如下：void resiz
- 前言：文章利用Python pygame做一个贪吃蛇的小游戏而且讲清楚每一段代码是用来干嘛的。据说是贪吃蛇游戏是1976年，Gremlin公
- python模块导入细节本文主要介绍了关于python模块导入的相关内容，分享出来供大家参考学习，下面话不多说了，来一起看看详细的介绍吧官方
- 提到SQL Server 2005证书，很多人可能以为它只是用来在传输数据的时候起到加密作用的，但在深入了解后，你会发现它的用处还有很多。
