trainer.py 文件源码

def evaluate(self):
        test_iter = chainer.iterators.SerialIterator(self.testset, 1,
                                                     repeat=False, shuffle=False)
        self.chain.train = False
        self.chain.test = True
        if self.gpu >= 0:
            self.chain.to_gpu(self.gpu)
        result = extensions.Evaluator(test_iter, self.chain, device=self.gpu)()
        if self.gpu >= 0:
            self.chain.to_cpu()
        #for k,v in result.iteritems():
        #    if k in ["main/numsamples", "main/accuracy", "main/branch0exit", "main/branch1exit", "main/branch2exit"]:
        #        print k, "\t\t\t", v
        return result

    # Deprecated