LazyGuard

class paddle. LazyGuard [源代码]

LazyGuard 是一个用于设置模型(继承自 paddle.nn.Layer ) 中参数延迟初始化的上下文管理器。配合使用 python 的 with 语句来将 with LazyGuard(): 代码块下所有模型在实例化时,其内部的参数均不会立即申请内存空间。

代码示例

>>> from paddle import LazyGuard
>>> from paddle.nn import Linear

>>> with LazyGuard():
...     # w and b are initialized lazily and have no memory.
...     net = Linear(10, 10)
...
>>> for param in net.parameters():
...     # Initialize param and allocate memory explicitly.
...     param.initialize()