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()

         