crbm.py 文件源码

def build_init_ops(self):
        if self.model_exists:
            self.init_ops = tf.initialize_local_variables()
        else:
            self.init_ops = [tf.initialize_local_variables(), tf.initialize_variables(self.variables)]
        self.saver = tf.train.Saver(self.variables)