提交 840a2eb0 编辑于 作者: Toshihiro Nakae's avatar Toshihiro Nakae
浏览文件

create graph explicitly in restore

上级 962939d3

dagmm/dagmm.py

+6 −4
原始行号 差异行号 差异行
@@ -176,7 +176,9 @@ class DAGMM: 
        model_path = join(fdir, self.MODEL_FILENAME)

        meta_path = model_path + ".meta"



        self.sess = tf.Session()

        with tf.Graph().as_default() as graph:

            self.graph = graph

            self.sess = tf.Session(graph=graph)

            self.saver = tf.train.import_meta_graph(meta_path)

            self.saver.restore(self.sess, model_path)