Tensorflow/Keras 并行训练多个模型实例
发布时间
阅读量：
阅读量
经过长时间的探索，最终成功达成了多线程环境下同步训练多个模型的目标。关键点在于深入掌握TF框架中的Graph与Session机制。无需过多解释，直接展示代码示例，通过阅读代码即可清晰理解实现方式！
class MyModel(object):
def __init__(self):
self.model1Thread = None
self.model2Thread = None
self.model1_graph = tf.Graph()
self.model1_sess = tf.Session(graph=self.model1_graph)
self.model2_graph = tf.Graph()
self.model2_sess = tf.Session(graph=self.model2_graph)
self.build_model1()
self.build_model2()
def build_model1(self):
with self.model1_sess.as_default():全部评论 (0)
还没有任何评论哟~