tensorflow 在同一个GPU同时加载多张相同的图

saver = self.tf_instance.train.Saver()
self.sess = self.tf_instance.Session(config=sess_config, graph=graph)
self.sess.run(self.tf_instance.global_variables_initializer())
# 在restore时加入这行代码
self.tf_instance.reset_default_graph()
saver.restore(self.sess, ckpt_file)

 

posted @ 2019-02-16 17:21  下路派出所  阅读(684)  评论(0编辑  收藏  举报