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¶