o ûp i½ã@shddlmZddlmZddlmZerddlmZdgZGdd„dƒZ e ƒa d d „Z Gd d„dƒZ d S) é)Ú annotations)Ú TYPE_CHECKINGé)Ú framework)Ú TracebackTypeÚ LazyGuardc@sDeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Ze d d „ƒZ dS)ÚLazyInitHelperz‡ A Helper Context to trigger switching mode between dygraph and static graph mode, and holds the startup program resource. cCsd|_d|_d|_dS)NF)Ú_stateÚ_tracerÚ _in_guard©Úself©rúf/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/nn/initializer/lazy_init.pyÚ__init__"s zLazyInitHelper.__init__cCs$|jrdSt ¡s Jdƒ‚d|_dS)zs Switch into lazy mode. NOTE(dev): This is a very low level API and not exposed for user. Nz4LazyInit.enable() is only available in dygraph mode.T)r rZin_dygraph_moder rrrÚenable's  ÿ zLazyInitHelper.enablecCs|jsdSd|_dS)zq Exit from lazy mode. NOTE(dev): This is a very low level API and not exposed for user. NF©r r rrrÚdisable4s zLazyInitHelper.disablecCs.| ¡|jr dStjj|_dtj_d|_dS)z‚ Switch into lazy mode and set _dygraph_tracer_ with None to convert dygraph mode into static graph mode. NT)rr rÚ global_varÚ_dygraph_tracer_r r rrrÚ __enter__>s   zLazyInitHelper.__enter__cOs:| ¡|js dS|jdusJ‚|jtj_d|_d|_dS)zC Exit from lazy mode and recover _dygraph_tracer_. NF)rr r rrr)r ÚargsÚkwargsrrrÚ__exit__Js  zLazyInitHelper.__exit__cCs|jS©Nrr rrrÚstateVszLazyInitHelper.stateN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrÚpropertyrrrrrrs    rcCstSr)Ú_lazy_init_helperrrrrÚlazy_init_helper^sr"c@s$eZdZdZddd„Zdd d „ZdS)ra± LazyGuard is a wrapper interface for nn.Layer, it forwards the construct process of user defined Layer. Meanwhile, it provides necessary API to trigger EagerParamBase Lazy Initialization and get startup Program. Examples: .. code-block:: python >>> 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() ÚreturnÚNonecCstƒ ¡dS)a¼ Construct instance from class_obj by Lazy Initializing parameters. Examples: .. code-block:: python >>> from paddle import LazyGuard >>> from paddle.nn import Linear >>> with LazyGuard(): ... fc = LazyInit(Linear)(10, 10) ... >>> for param in fc.parameters(): ... param.initialize() N)r"rr rrrryszLazyGuard.__enter__Úexc_typeútype[BaseException] | NoneÚexc_valúBaseException | NoneÚexc_tbúTracebackType | NonecCstƒ ¡dSr)r"r)r r%r'r)rrrrŒszLazyGuard.__exit__N)r#r$)r%r&r'r(r)r*r#r$)rrrrrrrrrrrcs N) Ú __future__rÚtypingrÚbaserÚtypesrÚ__all__rr!r"rrrrrÚs    ?