athena.deploy_main

a sample implementation of freezing model

Module Contents

Functions

freeze_session(session, output_names=None, clear_devices=True) build frozen graph
freeze_graph(json_file, frozen_graph_dir) freeze TensorFlow model trained on Python to pb format,
restore_model(json_file, frozen_graph_dir) restore the best model
athena.deploy_main.freeze_session(session, output_names=None, clear_devices=True)

build frozen graph

athena.deploy_main.freeze_graph(json_file, frozen_graph_dir)

freeze TensorFlow model trained on Python to pb format, which gather graph defination and weights together into a single file .

athena.deploy_main.restore_model(json_file, frozen_graph_dir)

restore the best model

athena.deploy_main.json_file