ϑi2cSSKJr SSKJr SSKrSSKJr SSKJr SSKJ r SSK J r SSK Jr SS KJrJrJr SS KJr \(aSS KJrJr SS KJr SS KJr SSKJrJr /r"SS\5rg)) annotations) TYPE_CHECKINGN)_C_ops) framework)base) LayerHelper)signature_safe_contextmanager)in_dynamic_modein_dynamic_or_pir_mode in_pir_mode) Optimizer) GeneratorSequence)Tensor)_ParameterConfig)ExecutorProgramc^\rSrSr%SrS\S'S\S'S\S'S\S 'S \S 'S \S 'S \S'SSU4SjjjrSrSr\ RSSSjj5r \ R\ RSSj55r\\ RSSSjj55r\ RS S!Sjj5rSrSrSrU=r$)" ModelAverage*a The ModelAverage optimizer accumulates specific continuous historical parameters during training. The accumulated historical range can be controlled by the passed ``average_window_rate`` argument. The averaged ``Parameter`` are used in the prediction, which usually can improve the accuracy of the prediction. Accumulate the average of the ``Parameter`` in the sliding window, the result will be saved in a temporary variable, can be applied to the current model's ``Parameter`` by calling the ``apply()`` method, and the current model ``Parameter`` can be restored by calling the ``restore()`` method. The window size for calculating the average is determined by ``average_window_rate``, ``min_average_window``, ``max_average_window`` and the current ``Parameter`` update times (num_updates). When the cumulative times (num_accumulates) is greater than the specific window threshold (average_window), the accumulated ``Parameter`` temporary variable is set to 0.0. The following example will help to understand the role of these arguments: :: if num_accumulates >= min_average_window and num_accumulates >= min(max_average_window, num_updates * average_window_rate): num_accumulates = 0 In the above conditional judgment statement, ``num_accumulates`` indicates the current accumulated number, which can be abstractly understood as the length of the cumulative window. The length of the window must be at least the length set by the ``min_average_window`` argument, and cannot exceed the length specified by the ``max_average_window`` argument or ``num_updates * average_window_rate``, where ``num_updates`` indicates the current ``Parameter`` update times, ``average_window_rate`` is a coefficient that calculates the length of the window. Args: average_window_rate (float): The calculate ratio of the window length relative to ``Parameter`` update times. parameters (list, optional): List of ``Tensor`` names to update to minimize ``loss``. \ This parameter is required in dygraph mode. \ The default value is None in static graph mode, at this time all parameters will be updated. min_average_window (int, optional): the minimum size of average window length. The default value is 10000. max_average_window (int, optional): The maximum size of average window length. The default value is 10000. name (str, optional): Normally there is no need for user to set this property. For more information, please refer to :ref:`api_guide_Name`. The default value is None. Examples: .. code-block:: python >>> # doctest: +SKIP("Cannot get source code by to_static in REPL") >>> import numpy as np >>> import paddle >>> import paddle.nn as nn >>> import paddle.optimizer as opt >>> BATCH_SIZE = 16 >>> BATCH_NUM = 4 >>> EPOCH_NUM = 4 >>> IMAGE_SIZE = 784 >>> CLASS_NUM = 10 >>> # define a random dataset >>> class RandomDataset(paddle.io.Dataset): # type: ignore[type-arg] ... def __init__(self, num_samples): ... self.num_samples = num_samples ... def __getitem__(self, idx): ... image = np.random.random([IMAGE_SIZE]).astype('float32') ... label = np.random.randint(0, CLASS_NUM - 1, (1, )).astype('int64') ... return image, label ... def __len__(self): ... return self.num_samples ... >>> class LinearNet(nn.Layer): ... def __init__(self): ... super().__init__() ... self._linear = nn.Linear(IMAGE_SIZE, CLASS_NUM) ... self.bias = self._linear.bias ... ... @paddle.jit.to_static ... def forward(self, x): ... return self._linear(x) ... >>> def train(layer, loader, loss_fn, opt, model_average): ... for epoch_id in range(EPOCH_NUM): ... for batch_id, (image, label) in enumerate(loader()): ... out = layer(image) ... loss = loss_fn(out, label) ... loss.backward() ... opt.step() ... model_average.step() ... opt.clear_grad() ... model_average.clear_grad() ... print("Train Epoch {} batch {}: loss = {}, bias = {}".format( ... epoch_id, batch_id, np.mean(loss.numpy()), layer.bias.numpy())) ... >>> def evaluate(layer, loader, loss_fn): ... for batch_id, (image, label) in enumerate(loader()): ... out = layer(image) ... loss = loss_fn(out, label) ... loss.backward() ... print("Evaluate batch {}: loss = {}, bias = {}".format( ... batch_id, np.mean(loss.numpy()), layer.bias.numpy())) ... >>> # create network >>> layer = LinearNet() >>> loss_fn = nn.CrossEntropyLoss() >>> optimizer = opt.Momentum(learning_rate=0.2, momentum=0.1, parameters=layer.parameters()) >>> model_average = paddle.incubate.ModelAverage( ... 0.15, ... parameters=layer.parameters(), ... min_average_window=2, ... max_average_window=10 ... ) ... >>> # create data loader >>> dataset = RandomDataset(BATCH_NUM * BATCH_SIZE) >>> loader = paddle.io.DataLoader(dataset, ... batch_size=BATCH_SIZE, ... shuffle=True, ... drop_last=True, ... num_workers=2) ... >>> # create data loader >>> eval_loader = paddle.io.DataLoader(dataset, ... batch_size=BATCH_SIZE, ... shuffle=True, ... drop_last=True, ... num_workers=1 ... ) ... >>> # train >>> train(layer, loader, loss_fn, optimizer, model_average) >>> print("\nEvaluate With ModelAverage") >>> with model_average.apply(need_restore=False): ... evaluate(layer, eval_loader, loss_fn) >>> print("\nEvaluate With Restored Parameters") >>> model_average.restore() >>> evaluate(layer, eval_loader, loss_fn) rhelperfloataverage_windowintmin_average_windowmax_average_windowstrtyper apply_programrestore_programc>[T U]SUSSUS9 [URR5UlXlX0lX@lSUl [5(Gd[RR5R5nU(aUOUR5nUR!Xg5 UHnUR#XhS/5 M [RR%5UlUR&R5n [RR)UR&S9 UHnUR+X5 M SSS5 [RR%5UlUR,R5n [RR)UR,S9 UHnUR/X5 M SSS5 gg!,(df  N=f!,(df  g=f)Ng) learning_rate parameters weight_decay grad_clipnameaverage_accumulates) main_program)super__init__r __class____name__rrrrrr paddlestaticdefault_main_program global_blockall_parameters_create_accumulators_append_optimize_oprr program_guard_add_average_apply_opr _add_average_restore_op) selfaverage_window_rater#rrr&r0r1paramblockr+s f/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/incubate/optimizer/modelaverage.pyr*ModelAverage.__init__s !  "$.."9"9: 1"4"4)   !====?LLNL( l.I.I.K   % %l C'((t}E(!'!6!6!8D &&335E,,$:L:L,M+E..u<,N$*==#8#8#:D ((557E,,$:N:N,O+E00>,PO!!NM POs6G"=G3" G03 Hc[U[R[RR45(deUHnUR SU5 UR SU5 UR SU5 UR SU5 UR SUSS/S9 UR S USS/S9 UR S USS/S9 M g) Nsum_1sum_2sum_3restorenum_accumulatesint64)dtypeshapeold_num_accumulates num_updates) isinstancerBlockr-pir_add_accumulator)r7r:r#r9s r;r2!ModelAverage._create_accumulatorss%)//6::3C3C!DEEEEE  ! !'5 1  ! !'5 1  ! !'5 1  ! !)U 3  ! !!5s "   ! !%uGA3 "   ! !uGA3 "  c [U[R[RR45(deUR SUS5nUR SUS5nUR SUS5nUR SUS5nUR SUS5nUR SUS5n[ 5(aH[R"USUUUUUUURURUR5 u n g[R"5R5nURURURS.n USUUUUUUS .n UUUUUUS .n URURU U U S S 9n U $) Nr>rr?r@rBrGrH)rrr)r9in_sum_1in_sum_2in_sum_3in_num_accumulatesin_old_num_accumulatesin_num_updates) out_sum_1 out_sum_2 out_sum_3out_num_accumulatesout_old_num_accumulatesout_num_updatesT)rinputsoutputsattrs stop_gradient)rIrrJr-rK_get_accumulatorr raverage_accumulates_rrrr/r0 append_opr)r7r:param_and_gradr>r?r@rBrGrH_r^r\r]average_accumulates_ops r;r3 ModelAverage._append_optimize_ops%)//6::3C3C!DEEEE%%g~a/@A%%g~a/@A%%g~a/@A// ~a0 #33 !>!#4 ++M>!;LM ! # #%::q!###''''  Aq!Q1..0==?"11"&"9"9"&"9"9 $A&"1&9) #2':*  "' "1" &%rNcD[5(aUR5 gg)a^ Add operations to minimize ``loss`` by updating ``parameters``. Args: loss (Tensor): A ``Tensor`` containing the value to minimize. startup_program (Program, optional): :ref:`api_paddle_static_Program` for initializing parameters in ``parameters``. The default value is None, at this time :ref:`api_paddle_static_default_startup_program` will be used. parameters (list, optional): List of ``Tensor`` or ``Tensor.name`` to update to minimize ``loss``. The default value is None, at this time all parameters will be updated. no_grad_set (set, optional): Set of ``Tensor`` or ``Tensor.name`` that don't need to be updated. The default value is None. Returns: tuple: tuple (optimize_ops, params_grads), A list of operators appended by minimize and a list of (param, grad) tensor pairs, param is ``Parameter``, grad is the gradient value corresponding to the parameter. In static graph mode, the returned tuple can be passed to ``fetch_list`` in ``Executor.run()`` to indicate program pruning. If so, the program will be pruned by ``feed`` and ``fetch_list`` before run, see details in ``Executor``. Examples: .. code-block:: python >>> import paddle >>> inp = paddle.rand([1, 10], dtype="float32") >>> linear = paddle.nn.Linear(10, 1) >>> out = linear(inp) >>> loss = paddle.mean(out) >>> loss.backward() >>> sgd = paddle.optimizer.SGD(learning_rate=0.1,parameters=linear.parameters()) >>> sgd.minimize(loss) >>> modelaverage = paddle.incubate.ModelAverage( ... 0.15, ... parameters=linear.parameters(), ... min_average_window=2, ... max_average_window=4 ... ) >>> modelaverage.minimize(loss) >>> sgd.clear_grad() >>> modelaverage.clear_grad() N)r step)r7lossstartup_programr# no_grad_sets r;minimizeModelAverage.minimize<sn    IIK rNcn/nURHKnUR(dMUR5cM)UR5nURX#45 MM [R "5R 5nURX@R5 UHnURXE5 M g)a Execute the optimizer and update parameters once. Returns: None Examples: .. code-block:: python >>> import paddle >>> inp = paddle.rand([1, 10], dtype="float32") >>> linear = paddle.nn.Linear(10, 1) >>> out = linear(inp) >>> loss = paddle.mean(out) >>> sgd = paddle.optimizer.SGD(learning_rate=0.1,parameters=linear.parameters()) >>> modelaverage = paddle.incubate.ModelAverage( ... 0.15, ... parameters=linear.parameters(), ... min_average_window=2, ... max_average_window=4 ... ) >>> loss.backward() >>> sgd.step() >>> modelaverage.step() >>> sgd.clear_grad() >>> modelaverage.clear_grad() N) _parameter_list trainable _grad_ivarappendrr/r0r2r3)r7 params_gradsr9grad_varr:rcs r;rhModelAverage.stepvs@ ))E??!- ++-##U$56 *..0==? !!%)=)=>*N  $ $U ;+rNc#2# [5(GaURHnURSU5nURSU5nURSU5nURSU5nURSU5nURSU5n [R"X95 Xg-U-n XE-n [R "U SS9n [R "U SS9n X- n [R"X5 M S v U(aUR 5 g Uc [S 5eURUR5 S v U(aUR U5 g g !U(aUR 5 ff=f!U(aUR U5 ff=f7f) a Apply the average of the cumulative ``Parameter`` to the parameters of the current model. Args: executor(Executor): The network executor in static-graph mode. The default value is None in dygraph mode. need_restore(bool): Restore flag variable, if set to True, the network will restore the parameters of the network to the default value, if set to False, it will not be restored. The default value is True. Examples: .. code-block:: python >>> import paddle >>> inp = paddle.rand([1, 10], dtype="float32") >>> linear = paddle.nn.Linear(10, 1) >>> out = linear(inp) >>> loss = paddle.mean(out) >>> loss.backward() >>> sgd = paddle.optimizer.SGD(learning_rate=0.1,parameters=linear.parameters()) >>> modelaverage = paddle.incubate.ModelAverage( ... 0.15, ... parameters=linear.parameters(), ... min_average_window=2, ... max_average_window=4 ... ) >>> sgd.step() >>> modelaverage.step() >>> with modelaverage.apply(): ... for param in linear.parameters(): ... print(param) >>> for param in linear.parameters(): ... print(param) rBrGr>r?r@rAfloat32)rEN1Executor should not be None in static graph mode.) r ror`r-assigncastrA RuntimeErrorrunr) r7executor need_restorer9rBrGr>r?r@ param_restore total_paramtotal_accumulates average_params r;applyModelAverage.applys{V   --"&"7"7%u#'+&;&;)5'#--gu=--gu=--gu= $ 5 5i G  e3#me3 $3$I!$kk+YG $*KK%Y%!!, ?  m3).* #LLN   C   T''( '  X& LLN  X&s7C5F8EE9FE66F9FFc[5(a<URH+nURSU5n[R"X25 M- gUc [ S5eUR UR5 g)a\ Restore ``Parameter`` values of current model. Args: executor(Executor): The network executor in static-graph mode. The default value is None in dygraph mode Examples: .. code-block:: python >>> import paddle >>> inp = paddle.rand([1, 10], dtype="float32") >>> linear = paddle.nn.Linear(10, 1) >>> out = linear(inp) >>> loss = paddle.mean(out) >>> loss.backward() >>> sgd = paddle.optimizer.SGD(learning_rate=0.1,parameters=linear.parameters()) >>> modelaverage = paddle.incubate.ModelAverage( ... 0.15, ... parameters=linear.parameters(), ... min_average_window=2, ... max_average_window=4 ... ) >>> sgd.step() >>> modelaverage.step() >>> with modelaverage.apply(need_restore=False): ... for param in linear.parameters(): ... print(param) >>> for param in linear.parameters(): ... print(param) >>> modelaverage.restore() >>> for param in linear.parameters(): ... print(param) rANrx)r ror`r-ryr{r|r )r7r}r9rs r;rAModelAverage.restoreskT   -- $ 5 5i G  m3.   C   T))*rNc[5(Ga[RR5n[RR R X25nURSU5n[RR RX45nURSU5n[RR RX65nURSU5n[RR RX75nURSU5n[RR RX85nURSU5n [RR RX95n URSU5n [RR RX:5n OURU5nURURSU55nURURSU55nURURSU55nURURSU55nURURSU55n URURSU55n [R"X%S9 [R"X/5n [R"XgU/5n [R"XRcSO URS 9n [R"XRcSO URS 9n [R"XS 9n [R"XS9 g) NrAr>r?r@rBrGoutputrw)xrE)ry)r r-r.r/rKcore_get_parameterr`_get_persistable_value_clone_variableryadd_nrz_dtypedivide)r7r:r9target_program restore_valuegradr>r?r@rBrGtmpsum divide_outs r;r5"ModelAverage._add_average_apply_op*s ==#]]??ANJJOO22>IE 11)UCM::??99D))'59EJJOO::E))'59EJJOO::E))'59EJJOO::E#334EuMO$jjooDDO#'"7"7%u# #)**//"H"H# ))%0E((%%i7D))$*?*?*OPE))$*?*?*OPE))$*?*?*OPE#33%%&7?O#("7"7%%&;UC#   e)llOABllE%01kkkk&9t{{ kkkk&9t{{ ]]S0  j/rNc[5(a[RR5n[RR R X25nURSU5n[RR RX45nO2URU5nURURSU55n[R"XRS9 g)NrAr) r r-r.r/rKrrr`rrry)r7r:r9rrrs r;r6$ModelAverage._add_average_restore_opds ==#]]??ANJJOO22>IE 11)UCM::??99D))%0E((%%i7D  d)rN)rrrrrr r)N'rN) r8rr#z4Sequence[Tensor] | Sequence[_ParameterConfig] | Nonerrrrr&z str | NonereturnNone)NNN) rirrjzProgram | Noner#zlist[Tensor] | Nonerkzset[Tensor] | set[str] | Nonerr)rr)NT)r}Executor | Noner~boolrzGenerator[None, None, None])N)r}rrr)r, __module__ __qualname____firstlineno____doc____annotations__r*r2r3imperative_baseno_gradrlr dygraph_onlyrhr rrAr5r6__static_attributes__ __classcell__)r+s@r;rr*sJX  I LP"'"' (?"(?I(? (?  (?  (? (?(?T$?&B+/*.59 77(7( 7 3 7  77r)<)BN' $N'#N'`2+2+h80t * *rNr) __future__rtypingrr-r paddle.baserpaddle.base.dygraphrrpaddle.base.layer_helperrpaddle.base.wrapped_decoratorr paddle.frameworkr r r paddle.optimizerr collections.abcrrrpaddle.optimizer.optimizerr paddle.staticrr__all__rrNr;rsR# !70G '3;/ G *9G *rN