繁体   English   中英

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

[英]How to restore the session from keras training model

我想从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训练模型中恢复会话?

最好的祝福。

已经有一些与您类似的已回答问题,我建议您先用Google对其进行搜索。 但是,如果他们没有工作,也许你可以试试一个,并试图挽救你的模型,并在另一个会话中使用它。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM