tensorflow的保存模型大小理解（生成模型和打印模型参数）-20201110

最新推荐文章于 2024-03-23 14:29:05 发布

原创

最新推荐文章于 2024-03-23 14:29:05 发布 · 1.6k 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍如何使用TensorFlow创建并保存模型，重点展示了不同优化器对模型文件大小的影响，并提供了检查模型参数的方法。

tensorflow的保存模型大小理解过程

如何计算tensorflow计算model的大小size（1）
如何计算tensorflow计算model的大小size（2）
https://github.com/Prinsphield/Wechat_AutoJump/issues/56
https://stackoverflow.com/questions/39757985/why-are-there-two-additional-variables-in-the-checkpoint-for-each-layer（这篇说明Adam3倍原因，说实话，还不是很懂）
1.生成模型

import os
import tensorflow as tf
import tensorflow.contrib.slim as slim
 
CURRENT_DIR = os.getcwd()
train_dir  = CURRENT_DIR + '/logs/'
 
a = tf.placeholder(tf.float32, shape=[None, 6, 512], name='a')
b = slim.fully_connected(a, 1 * 512, activation_fn=tf.nn.relu, weights_initializer=slim.variance_scalin