Question

我想从keras训练模型中恢复会话。

我试图通过以下过程将其还原。

1，创建由keras训练的检查点文件

from keras.models import *
import keras.backend as K
from keras.applications.vgg16 import VGG16

model = VGG16_convolutions()
model.fit_generator(...)

with tf.Session() as ksess:
ksess = K.get_session()
saver.save(ksess, "./ksess.cpkt", global_step=0, latest_filename="checkpoint_state")

2，恢复张量流的会话

import tensorflow as tf

with tf.name_scope("block1_conv1"):
    block1_conv1_kernel = tf.Variable(initial_value=0, name="kernel")
    block1_conv1_bias = tf.Variable(initial_value=0, name="bias")
with tf.name_scope("block1_conv2"):
    block1_conv2_kernel = tf.Variable(initial_value=0, name="kernel")
    block1_conv2_bias = tf.Variable(initial_value=0, name="bias")
...

sess = tf.Session()
saver = tf.train.Saver()
saver = tf.train.import_meta_graph("ksess.ckpt-0.meta")
saver.restore(sess, "./ksess.ckpt")

那么，如何从keras训练模型中恢复会话？

最诚挚的问候。

Answer 1

已经有一些与您类似的已回答问题，我建议您先用Google对其进行搜索。但是，如果它们都不起作用，也许您可以尝试this并尝试保存模型并在另一个会话中使用它。

如何从keras训练模型中恢复会话

1 个答案: