恢复模型tens

2024-04-25 19:18:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在测试保存模型,然后在另一个python实例中恢复它的选项。为此,我将使用以下代码保存模型:

saver = tf.train.Saver([weights0,weights1,weights2,syn0,bias0,syn1,bias1],max_to_keep = 2)
saver.save(sess, '..\\places365_TFG\\models\\CNNFinalModel', global_step=l)

在python的另一个实例中,我这样做:

new_saver = tf.train.import_meta_graph('..\\places365_TFG\\models\\CNNFinalModel-67.meta')
new_saver.restore(sess, 
tf.train.latest_checkpoint('..\\places365_TFG\\models\\./'))
graph = tf.get_default_graph()
weights0 = graph.get_tensor_by_name("weights0:0")

要检查weights0的值是否恢复得很好,我需要:

sess.run(weights0)

变量weights0的初始化方式如下:

def weight_variable(shape,name):
   initial = tf.truncated_normal(shape, stddev=0.1, name=name)
   return tf.Variable(initial)

weights0 = weight_variable([3,3,3,16], name='weights0')

我看到的问题是,每次我执行这个,weighsts0的值都会改变。我想这是因为它被定义为变量,但我想保留我从原始神经网络中保存的值。你知道吗

我该怎么做?你知道吗

先谢谢你。你知道吗


Tags: 实例name模型newmodelstftrainmeta