(unused_args, flags)
| 362 | |
| 363 | |
| 364 | def main(unused_args, flags): |
| 365 | if flags.checkpoint_version == 1: |
| 366 | checkpoint_version = saver_pb2.SaverDef.V1 |
| 367 | elif flags.checkpoint_version == 2: |
| 368 | checkpoint_version = saver_pb2.SaverDef.V2 |
| 369 | else: |
| 370 | raise ValueError("Invalid checkpoint version (must be '1' or '2'): %d" % |
| 371 | flags.checkpoint_version) |
| 372 | freeze_graph(flags.input_graph, flags.input_saver, flags.input_binary, |
| 373 | flags.input_checkpoint, flags.output_node_names, |
| 374 | flags.restore_op_name, flags.filename_tensor_name, |
| 375 | flags.output_graph, flags.clear_devices, flags.initializer_nodes, |
| 376 | flags.variable_names_whitelist, flags.variable_names_blacklist, |
| 377 | flags.input_meta_graph, flags.input_saved_model_dir, |
| 378 | flags.saved_model_tags, checkpoint_version) |
| 379 | |
| 380 | |
| 381 | def run_main(): |
no test coverage detected