Python is_grad_enabled示例

示例#1

文件： test_autograd_mode.py 项目： ZJLabDubhe/oneflow-zj

    def test_no_grad(test_case):
        with flow.no_grad():
            test_case.assertFalse(flow.is_grad_enabled())
        test_case.assertTrue(flow.is_grad_enabled())

        @flow.no_grad()
        def func():
            test_case.assertFalse(flow.is_grad_enabled())

        func()
        test_case.assertTrue(flow.is_grad_enabled())

示例#2

文件： test_autograd_mode.py 项目： zzk0/oneflow

    def test_grad_enable(test_case):
        with flow.grad_enable():
            test_case.assertTrue(flow.is_grad_enabled())
        test_case.assertTrue(flow.is_grad_enabled())

        @flow.grad_enable()
        def func():
            test_case.assertTrue(flow.is_grad_enabled())

        func()
        test_case.assertTrue(flow.is_grad_enabled())

示例#3

文件： test_autograd_mode.py 项目： ZJLabDubhe/oneflow-zj

    def test_set_grad_enabled(test_case):
        with flow.set_grad_enabled(True):
            test_case.assertTrue(flow.is_grad_enabled())
        test_case.assertTrue(flow.is_grad_enabled())

        @flow.set_grad_enabled(True)
        def func():
            test_case.assertTrue(flow.is_grad_enabled())

        func()
        test_case.assertTrue(flow.is_grad_enabled())

        with flow.set_grad_enabled(False):
            test_case.assertFalse(flow.is_grad_enabled())
        test_case.assertTrue(flow.is_grad_enabled())

        @flow.set_grad_enabled(False)
        def func():
            test_case.assertFalse(flow.is_grad_enabled())

        func()
        test_case.assertTrue(flow.is_grad_enabled())

示例#4

文件： test_autograd_mode.py 项目： ZJLabDubhe/oneflow-zj

    def test_inference_mode(test_case):
        with flow.inference_mode(True):
            test_case.assertFalse(flow.is_grad_enabled())
        test_case.assertTrue(flow.is_grad_enabled())

        @flow.inference_mode(True)
        def func():
            test_case.assertFalse(flow.is_grad_enabled())

        func()
        test_case.assertTrue(flow.is_grad_enabled())

        with flow.inference_mode(False):
            test_case.assertTrue(flow.is_grad_enabled())
        test_case.assertTrue(flow.is_grad_enabled())

        @flow.inference_mode(False)
        def func():
            test_case.assertTrue(flow.is_grad_enabled())

        func()
        test_case.assertTrue(flow.is_grad_enabled())

示例#5

文件： test_autograd_mode.py 项目： ZJLabDubhe/oneflow-zj

 def func():
     test_case.assertFalse(flow.is_grad_enabled())

示例#6

文件： test_autograd_mode.py 项目： ZJLabDubhe/oneflow-zj

 def test_grad_mode(test_case):
     test_case.assertTrue(flow.is_grad_enabled())