tensorflow restore ckpt里有的参数，ckpt里没有不加载

以你之姓@ 2023-01-13 14:52 92阅读 0赞

def my_restore(session = tf.Session(), save_file = "model.ckpt-100000"):
        reader = tf.train.NewCheckpointReader(save_file)
        saved_shapes = reader.get_variable_to_shape_map()
        var_names = sorted([(var.name, var.name.split(':')[0]) for
                            var in tf.global_variables()
                            if var.name.split(':')[0] in saved_shapes])
        to_restore_variables = []
        name2var = dict(zip(map(lambda x: x.name.split(':')[0], tf.global_variables()), tf.global_variables()))
        with tf.variable_scope('', reuse=True):
            for var_name, saved_var_name in var_names:
                curr_var = name2var[saved_var_name]
                var_shape = curr_var.get_shape().as_list()
                if var_shape == saved_shapes[saved_var_name] and "global_step" not in saved_var_name:
                    to_restore_variables.append(curr_var)
                else:
                    print("variable not trained.var_name:", var_name)
        saver = tf.train.Saver(to_restore_variables)
        saver.restore(session, save_file)

tensorflow版本：1.14 or 1.15

发表评论取消回复

表情：

评论列表（有 0 条评论，92人围观）

还没有评论，来说两句吧...

相关阅读

相关 tensorflow 将ckpt保存为pb文件

from tensorflow.python.framework.graph_util import convert_variables_to_constants

秒速五厘米/ 2023年01月13日 14:52/ 0 赞/ 162 阅读

相关 tensorflow restore ckpt里有的参数，ckpt里没有不加载

def my_restore(session = tf.Session(), save_file = "model.ckpt-100000"): rea

以你之姓@/ 2023年01月13日 14:52/ 0 赞/ 93 阅读

相关获取TensorFlow ckpt模型里的变量名称以及对应的值

一：输出模型中所有变量及其对应的值 from tensorflow.python.tools.inspect_checkpoint import print_tens

╰+攻爆jí腚メ/ 2022年12月20日 01:28/ 0 赞/ 122 阅读

相关 ckpt转pb

有时需要将tensorflow训练得到的ckpt固化成pb使用。Tensorflow提供了相关的固化命令脚本，下面是我搜集到的转换步骤。参考https://www.yanxis

淩亂°似流年/ 2022年09月11日 00:22/ 0 赞/ 178 阅读

相关 Tensorflow加载预训练模型和保存模型（ckpt文件）以及迁移学习finetuning

使用tensorflow过程中，训练结束后我们需要用到模型文件。有时候，我们可能也需要用到别人训练好的模型，并在这个基础上再次训练。这时候我们需要掌握如何操作这些模型数据。看完

待我称王封你为后i/ 2022年09月04日 11:00/ 0 赞/ 522 阅读

相关 Tensorflow: 读取ckpt文件中的tensor

在使用pre-train model时候，我们需要restore variables from checkpoint files. 经常出现在checkpoint 中找不到

心已赠人/ 2022年05月20日 09:04/ 0 赞/ 186 阅读

相关 tensorflow框架.ckpt .pb模型节点tensor_name打印及ckpt模型转.pb模型

转换模型首先要知道的是从哪个节点输出，如果没有源代码是很难清楚节点信息。获取ckpt模型的节点名称 import os from tensorflow

ゝ一纸荒年。/ 2022年05月15日 12:20/ 0 赞/ 211 阅读

相关 tensorflow实现将ckpt转pb文件

tensorflow实现将ckpt转pb文件【[尊重原创，转载请注明出处][Link 1]】：[https://blog.csdn.net/guyuealian/art

﹏ヽ暗。殇╰゛Y/ 2022年05月13日 23:10/ 0 赞/ 217 阅读

相关 tensorflow中ckpt转pb模型

参考代码： import tensorflow as tf import os from tensorflow.python.framework import gra

朱雀/ 2022年01月22日 01:17/ 0 赞/ 251 阅读

相关 ckpt文件restore的两种方法

这里以BERT恢复模型为例，说明恢复ckpt的两种常见方法 \ 方法一： init_checkpoint = "chinese_L-12_H-768_A-12/be

朱雀/ 2021年09月28日 12:52/ 0 赞/ 794 阅读