45ca429994196d479bef22edbbf36dc9698350d2,tests/test_utils/test_model_utils.py,ModelUtilTester,test_model_tester_cuda,#ModelUtilTester#,14

Before Change

            mu.ModelTester((1, 10, 32, 32), (1, 30, 32, 32))(model)

    def test_model_tester_cuda(self):
        if not torch.cuda.is_available():
            return
        model = \
            mu.ModelTester((1, 10, 32, 32), (1, 20, 32, 32))(nn.Conv2d(10, 20, 3, padding=1).cuda())
        with self.assertRaises(ShapeError):
            mu.ModelTester((1, 10, 32, 32), (1, 30, 32, 32))(model)

After Change

    @unittest.skipUnless(torch.cuda.is_available())
    def test_model_tester_cuda(self):
        tester = mu.ModelTester((1, 10, 32, 32), (1, 20, 32, 32)).cuda()
        model = tester(nn.Conv2d(10, 20, 3, padding=1).cuda())
        with self.assertRaises(ShapeError):
            mu.ModelTester((1, 10, 32, 32), (1, 30, 32, 32))(model)

Instances