test.py 文件源码

def test_cifar10(self):
        inputs = Variable(torch.randn(1,3,32,32))
        f, params, stats = define_diracnet(34, 1, 'CIFAR10')
        outputs = f(inputs, params, stats, mode=False)
        self.assertEqual(outputs.size(), torch.Size((1, 10)))