Python 3.x 如何使用python在tensorflow中使用.ckpt扩展而不是.meta扩展保存神经网络模型?
我不熟悉tensorflow和python。我试图用Python 3.x 如何使用python在tensorflow中使用.ckpt扩展而不是.meta扩展保存神经网络模型?,python-3.x,tensorflow,Python 3.x,Tensorflow,我不熟悉tensorflow和python。我试图用.ckpt扩展保存我训练过的神经网络模型。为此,我使用了以下代码: with tf.Session() as sess: sess.run(tf.initialize_all_variables()) saver = tf.train.Saver() for epoch in range(hm_epochs): epoch_loss = 0 i =
.ckpt
扩展保存我训练过的神经网络模型。为此,我使用了以下代码:
with tf.Session() as sess:
sess.run(tf.initialize_all_variables())
saver = tf.train.Saver()
for epoch in range(hm_epochs):
epoch_loss = 0
i =0;
while i < len(train_x):
start = i
end = i+batch_size
batch_x = np.array(train_x[start:end])
batch_y = np.array(train_y[start:end])
_, c = sess.run([optimizer, cost], feed_dict={x: batch_x, y: batch_y})
epoch_loss += c
i += batch_size
print('Epoch', epoch+1, 'completed out of',hm_epochs,'loss:',epoch_loss)
correct = tf.equal(tf.argmax(prediction, 1), tf.argmax(y, 1))
accuracy = tf.reduce_mean(tf.cast(correct, 'float'))
print('Accuracy:',accuracy.eval({x:test_x, y:test_y}))
saver.save(sess, 'my-model')
将tf.Session()作为sess的:
sess.run(tf.initialize\u all\u variables())
saver=tf.train.saver()
对于范围内的历元(hm_历元):
历元损失=0
i=0;
当i
但这将使用
.meta
扩展保存模型。有人能帮我用.ckpt
扩展名保存模型吗 您的代码很好:您不需要使用.ckpt
扩展名保存检查点
检查点文件的版本1(V1)中仅使用了.ckpt
扩展名。
从Tensorflow0.12
可以读取检查点V1,但当您保存一个新的检查点时,它将使用V2格式,您选择的名称只是.meta
和.index
文件的前缀
如果确实需要编写
.ckpt
文件,则必须将Tensorflow安装降级为您的代码很好:不需要使用.ckpt
扩展名保存检查点
检查点文件的版本1(V1)中仅使用了.ckpt
扩展名。
从Tensorflow0.12
可以读取检查点V1,但当您保存一个新的检查点时,它将使用V2格式,您选择的名称只是.meta
和.index
文件的前缀
如果确实需要编写.ckpt
文件,则必须将Tensorflow安装降级为