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

Question

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

我尝试通过以下过程恢复它。

1、创建keras训练的checkpoint文件

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、恢复tensorflow的session

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

已经有一些与你类似的问题得到解答，我建议你先用谷歌搜索。但如果它们都不起作用，也许您可​​以尝试this 并尝试保存您的模型并在另一个会话中使用它。