如何重用循环中创建的Tensorflow操作

时间:2018-06-04 01:07:37

标签: python tensorflow

我有一个在循环中创建多个优化器的方法。多次调用此方法,并且每次调用方法时我都不想重新创建这些优化器。有没有办法获得以前创建的优化器,以便我不必创建新的优化器?

def trainWordEmbeddings(self, sent_words, sent_num, window_size):
        for j in range(len(sent_words)):
            pos_context = self.getPosContext(sent_words, j, window_size)
            neg_context = self.neg_contexts.getNegContext(j, sent_num, pos_context)
            word_obj = self.objective.objective(WORD, sent_words[j], pos_context, neg_context)
            opt = tf.train.AdamOptimizer(learning_rate=self.eta).minimize(word_obj)
            self.session.run(opt)

0 个答案:

没有答案