// Freeze the models to protos.
freeze_graph("ml_perf/target/{}/target".format(N))
freeze_graph("ml_perf/checkpoint/{}/start".format(N))
finally:
asyncio.get_event_loop().close()
After Change
// Freeze the training checkpoint models.
pattern = os.path.join(FLAGS.dst_dir, "checkpoint", N, "work_dir", "*.index")
for path in glob.glob(pattern):
freeze_graph(os.path.splitext(path)[0])
finally:
asyncio.get_event_loop().close()