def save(self, ckpt_dir, version, values):
if len(self._meta) == 0:
return
keys = self._meta.keys()
values = [min(v.flatten().tolist()) for v in values]
assert len(keys) == len(values)
buf = json.dumps(dict(zip(keys, values)))
path = os.path.join(ckpt_dir, version, '.meta')
write_string_to_file(path, buf)
def export_graph(self, as_text=False):
for (op_name, input_name, input_type, size,
table) in graph_tag().inputs:
self.append_input(op_name, input_name, input_type, size, table)
self._graph_def.tag.output.op_name = graph_tag().output_op_name
if as_text:
path = os.path.join(self._ckpt_dir, "graph.txt")
write_string_to_file(path, str(self._graph_def))
else:
path = os.path.join(self._ckpt_dir, "graph.pb")
write_string_to_file(path, graph_pb.SerializeToString())
def export_graph(self, output_dir, as_text=False):
print 'start export graph'
for (op_name, input_name, input_type, size, table) in graph_tag().inputs:
self.append_input(op_name, input_name, input_type, size, table)
self._graph_def.tag.output.op_name = graph_tag().output_op_name
if as_text:
write_string_to_file("graph.txt", str(self._graph_def))
output(output_dir, "graph.txt")
else:
write_string_to_file("graph.pb", self._graph_def.SerializeToString())
output(output_dir, "graph.pb")
print 'finish export graph'
def save_meta(self, version, **kwargs):
kwargs['xdl_global_step'] = get_global_step().value
values = []
for k, v in kwargs.iteritems():
if isinstance(v, Tensor):
v = execute(v)
values.append(v)
else:
values.append(np.array(v))
values = [min(v.flatten().tolist()) for v in values]
keys = kwargs.keys()
assert len(keys) == len(values)
buf = json.dumps(dict(zip(keys, values)))
path = os.path.join(self._ckpt_dir, version, '.meta')
write_string_to_file(path, buf)
def save(self, path):
write_string_to_file(path, self.to_string())