sh:SSKrSSKJr S/rSr"SS\5rg)N) BaseSchedulerCubicSLc,[U[X 55$N)maxmin)xlohis ~/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/ao/pruning/scheduler/cubic_scheduler.py_clampr s r3r: cX^\rSrSrSrSU4Sjjr\SSj5rSrSr U=r $) raSets the sparsity level of each parameter group to the final sl plus a given exponential function. .. math:: s_i = s_f + (s_0 - s_f) \cdot \left( 1 - \frac{t - t_0}{n\Delta t} \right)^3 where :math:`s_i` is the sparsity at epoch :math:`t`, :math;`s_f` is the final sparsity level, :math:`f(i)` is the function to be applied to the current epoch :math:`t`, initial epoch :math:`t_0`, and final epoch :math:`t_f`. :math:`\Delta t` is used to control how often the update of the sparsity level happens. By default, Args: sparsifier (BaseSparsifier): Wrapped sparsifier. init_sl (int, list): Initial level of sparsity init_t (int, list): Initial step, when pruning starts delta_t (int, list): Pruning frequency total_t (int, list): Total number of pruning steps initially_zero (bool, list): If True, sets the level of sparsity to 0 before init_t (:math:`t_0`). Otherwise, the sparsity level before init_t (:math:`t_0`) is set to init_sl(:math:`s_0`) last_epoch (int): The index of last epoch. Default: -1. verbose (bool): If ``True``, prints a message to stdout for each update. Default: ``False``. c >XlURU5UlURU5UlURU5UlURU5UlURU5Ul[T U]!XU5 gr) sparsifier_make_sure_a_listinit_slinit_tdelta_ttotal_tinitially_zerosuper__init__) selfrrrrrr last_epochverbose __class__s r rCubicSL.__init__*st%--g6 ,,V4 --g6 --g6 "44^D 9rcfU(aX#:agXU- SX#- XE-- - S---n[XpU5nU$)a"Computes the current level of sparsity. Based on https://arxiv.org/pdf/1710.01878.pdf Args: s_0: Initial level of sparsity, :math:`s_i` s_f: Target level of sparsity, :math:`s_f` t: Current step, :math:`t` t_0: Initial step, :math:`t_0` dt: Pruning frequency, :math:`\Delta T` n: Pruning steps, :math:`n` initially_zero: Sets the level of sparsity to 0 before t_0. If False, sets to s_0 Returns: The sparsity level :math:`s_t` at the current step :math:`t` rg?)r)s_0s_ftt_0dtnrs_ts r sparsity_compute_fnCubicSL.sparsity_compute_fn@sC& ag3Y3!'bf)=#=!"CCCSs# rctUR(d[R"S5 [URUR UR URURUR5VVVVVVs/sH&upp4pVURUUURUUUUS9PM( snnnnnn$s snnnnnnf)NzUTo get the last sparsity level computed by the scheduler, please use `get_last_sl()`.)r#r$r%r&r'r(r) _get_sl_called_within_stepwarningswarnziprbase_slrrrrr*r)rinitial_sparsityfinal_sparsity initial_epoch delta_epochinterval_epochsrs r get_slCubicSL.get_slYs.. MM. ru     ## r  rm -o  $ $$"//!!- % r   s;-B0)rrrrrr)gr dFF)F) __name__ __module__ __qualname____firstlineno____doc__r staticmethodr*r7__static_attributes__ __classcell__)rs@r rrsB<:,0  r)r.base_schedulerr__all__rrrr rGs') +c mc r