(self)
| 653 | count += self.batch_size |
| 654 | |
| 655 | def evaluate(self): |
| 656 | config = tf.ConfigProto(allow_soft_placement=True) |
| 657 | with tf.Session(config=config) as sess: |
| 658 | with tf.device("/gpu:%d" % cfg.GPU_ID): |
| 659 | if self.model_path.find('.ckpt') != -1: |
| 660 | self.init_opt() |
| 661 | print("Reading model parameters from %s" % self.model_path) |
| 662 | saver = tf.train.Saver(tf.all_variables()) |
| 663 | saver.restore(sess, self.model_path) |
| 664 | # self.eval_one_dataset(sess, self.dataset.train, |
| 665 | # self.log_dir, subset='train') |
| 666 | self.eval_one_dataset(sess, self.dataset.test, |
| 667 | self.log_dir, subset='test') |
| 668 | else: |
| 669 | print("Input a valid model path.") |
no test coverage detected