no_grad

  • paddle.fluid.dygraph.no_grad(func)

在动态图模式中,此装饰器将会避免 func 被装饰时创建反向传播网络。

  • 参数:
    • func (str) – 不需要梯度的函数。

代码示例

  1. import numpy as np
  2. import paddle.fluid as fluid
  3.  
  4. @fluid.dygraph.no_grad
  5. def test_layer():
  6. with fluid.dygraph.guard():
  7. inp = np.ones([3, 32, 32], dtype='float32')
  8. t = fluid.dygraph.base.to_variable(inp)
  9. fc1 = fluid.FC('fc1', size=4, bias_attr=False, num_flatten_dims=1)
  10. fc2 = fluid.FC('fc2', size=4)
  11. ret = fc1(t)
  12. dy_ret = fc2(ret)
  13.  
  14. test_layer()