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())