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

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

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 one，并尝试保存模型并在另一个会话中使用它。