将模型参数打印出来——checkpoint模型、pb模型及其savemodel格式模型

最新推荐文章于 2024-08-21 09:53:35 发布

leap_ruo

最新推荐文章于 2024-08-21 09:53:35 发布

阅读量1.5k

点赞数 3

分类专栏：算法文章标签：深度学习 tensorflow

本文链接：https://blog.csdn.net/Airstudy/article/details/124675004

版权

本文介绍了在深度学习中，如何查看不同存储格式的模型参数，包括checkpoint、PB模型以及SaveModel形式的PB模型。在解决参数错误或需要探查模型参数时，提供了相应的读取操作方法。

摘要由CSDN通过智能技术生成

背景

有时候我们需要探查模型参数，或者在调用模型提示参数错误时，也需要查看模型参数来定位问题，以下是针对不同存储格式的模型，对参数做读取操作

checkpoint模型

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import tensorflow as tf
from tensorflow.python import pywrap_tensorflow


def show_ckp_params():
    model_dir = "./output/choose_model"
    ckpt = tf.train.get_checkpoint_state(model_dir)
    ckpt_path = ckpt.model_checkpoint_path
    reader = pywrap_tensorflow.NewCheckpointReader(ckpt_path)
    param_dict = reader.get_variable_to_shape_map()
    for key, val in param_dict