TensorFlow - Xuất
Ở đây, chúng ta sẽ tập trung vào việc hình thành MetaGraph trong TensorFlow. Điều này sẽ giúp chúng tôi hiểu mô-đun xuất trong TensorFlow. MetaGraph chứa thông tin cơ bản, được yêu cầu để đào tạo, thực hiện đánh giá hoặc chạy suy luận trên một biểu đồ đã được đào tạo trước đó.
Sau đây là đoạn mã cho cùng một -
def export_meta_graph(filename = None, collection_list = None, as_text = False):
"""this code writes `MetaGraphDef` to save_path/filename.
Arguments:
filename: Optional meta_graph filename including the path. collection_list:
List of string keys to collect. as_text: If `True`,
writes the meta_graph as an ASCII proto.
Returns:
A `MetaGraphDef` proto. """
Một trong những mô hình sử dụng điển hình cho điều tương tự được đề cập dưới đây:
# Build the model ...
with tf.Session() as sess:
# Use the model ...
# Export the model to /tmp/my-model.meta.
meta_graph_def = tf.train.export_meta_graph(filename = '/tmp/my-model.meta')