Method test_eval_mode

tests/networks/utils/test_train_mode.py:23–30

(self)

class TestEvalMode(unittest.TestCase):

    def test_eval_mode(self):
        t = torch.rand(1, 1, 4, 4)
        p = torch.nn.Conv2d(1, 1, 3)
        p.eval()
        self.assertFalse(p.training)  # False
        with train_mode(p):
            self.assertTrue(p.training)  # True
            p(t).sum().backward()


if __name__ == "__main__":