Example #1
0
    def init_evaluator(self, load_ckpt_path=""):
        if load_ckpt_path == "":
            load_ckpt_path = self.checkpoint_dir

        checkpoint = tf.train.latest_checkpoint(load_ckpt_path)

        q_values = self.graph_ops["q_values"]
        self.evaluator = tflearn.Evaluator(q_values, checkpoint)
def build_evaluator(trainer, target):

    evaluator = tflearn.Evaluator(target, session=trainer.session)

    return evaluator
Example #3
0
 def return_evaluator(self, trainer):
     evaluator = tflearn.Evaluator([self.mu, self.logvar],
                                   session=trainer.session)
     return evaluator
def define_evaluator(trainer, mu, logvar):
    evaluator = tflearn.Evaluator([mu, logvar], session=trainer.session)
    return evaluator