Sử dụng keras với lưu lượng tensorflow

Question

Tôi đã tự hỏi làm thế nào để sử dụng mô hình của tôi được đào tạo với keras trên một máy chủ sản xuất. Tôi nghe nói về phân phối lưu lượng, nhưng tôi không thể tìm ra cách sử dụng nó với mô hình keras của tôi.

Tôi thấy liên kết này: https://blog.keras.io/keras-as-a-simplified-interface-to-tensorflow-tutorial.html

Nhưng tôi không biết làm thế nào để khởi tạo biến sess, vì mô hình của tôi đã được đào tạo và tất cả mọi thứ. Có cách nào để thực hiện việc này không?

Answer 1

Bạn có thể khởi tạo biến phiên của bạn như

from keras import backend as K 
sess = K.get_session() 

và đi về xuất khẩu mô hình như trong hướng dẫn (Lưu ý rằng nhập khẩu đối với xuất khẩu đã thay đổi)

from tensorflow.contrib.session_bundle import exporter 

K.set_learning_phase(0) 
export_path = ... # where to save the exported graph 
export_version = ... # version number (integer) 

saver = tf.train.Saver(sharded=True) 
model_exporter = exporter.Exporter(saver) 
signature = exporter.classification_signature(input_tensor=model.input, 
               scores_tensor=model.output) 
model_exporter.init(sess.graph.as_graph_def(), 
        default_graph_signature=signature) 
model_exporter.export(export_path, tf.constant(export_version), sess)