i"$BddlmZmZmZddlmZeGddZeGddeZeGddeZeGd d eZ eGd d eZ eGd de Z eGdde Z eGddeZ eGddeZy))asdict dataclassfield)overridecNeZdZdZdefdZdeddfdZededdfdZd dZ y) _BaseProgressz>Mixin that implements state-loading utilities for dataclasses.returnct|SN)rselfs u/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pytorch_lightning/loops/progress.py state_dictz_BaseProgress.state_dicts d|rNc:|jj|yr )__dict__updater rs rload_state_dictz_BaseProgress.load_state_dicts Z(rc6|}|j||Sr )r)clsrobjs rfrom_state_dictz_BaseProgress.from_state_dictse J' rct)zReset the object's state.)NotImplementedErrorr s rresetz_BaseProgress.reset#s!!rr N) __name__ __module__ __qualname____doc__dictrr classmethodrrrrrrsKHD)$)4)/ "rrcXeZdZUdZdZeed<dZeed<ed dZ d dZ d eddfd Z y) _ReadyCompletedTrackeraFTrack an event's progress. Args: ready: Intended to track the number of events ready to start. completed: Intended to be incremented after the event completes (e.g. after ``on_*_end`` runs). These attributes should be increased in order, that is, :attr:`ready` first and :attr:`completed` last. rready completedr Nc d|_d|_y)zReset the state.rNr'r(r s rrz_ReadyCompletedTracker.reset7s rc&|j|_y)zReset the progress on restart. If there is a failure before all attributes are increased, restore the attributes to the last fully completed value. N)r(r'r s rreset_on_restartz'_ReadyCompletedTracker.reset_on_restart=s^^ rncX|xj|z c_|xj|z c_yr r*r r-s r increment_byz#_ReadyCompletedTracker.increment_byFs a  !rr) rrr r!r'int__annotations__r(rrr,r0r$rrr&r&(sFE3NIs  $cdrr&creZdZUdZdZeed<ed fd Zed fd Z ededdffd Z xZ S) _StartedTrackeraTrack an event's progress. Args: ready: Intended to track the number of events ready to start. started: Intended to be incremented after the event is started (e.g. after ``on_*_start`` runs). completed: Intended to be incremented after the event completes (e.g. after ``on_*_end`` runs). These attributes should be increased in order, that is, :attr:`ready` first and :attr:`completed` last. rstartedr Nc0t|d|_yNr)superrr5r __class__s rrz_StartedTracker.resetZs   rcDt||j|_yr )r8r,r(r5r9s rr,z _StartedTracker.reset_on_restart_s  "~~ rr-cNt|||xj|z c_yr )r8r0r5r r-r:s rr0z_StartedTracker.increment_byds Q  rr) rrr r!r5r1r2rrr,r0 __classcell__r:s@rr4r4Ks\ GS &&cdrr4creZdZUdZdZeed<ed fd Zed fd Z ededdffd Z xZ S) _ProcessedTrackeraTrack an event's progress. Args: ready: Intended to track the number of events ready to start. started: Intended to be incremented after the event is started (e.g. after ``on_*_start`` runs). processed: Intended to be incremented after the event is processed. completed: Intended to be incremented after the event completes (e.g. after ``on_*_end`` runs). These attributes should be increased in order, that is, :attr:`ready` first and :attr:`completed` last. r processedr Nc0t|d|_yr7)r8rrBr9s rrz_ProcessedTracker.resetzs  rcDt||j|_yr )r8r,r(rBr9s rr,z"_ProcessedTracker.reset_on_restarts  "rr-cNt|||xj|z c_yr )r8r0rBr=s rr0z_ProcessedTracker.increment_bys Q !rr) rrr r!rBr1r2rrr,r0r>r?s@rrArAjs\ Is ((cdrrAceZdZUdZeeZeed<eeZ eed<ddZ ddZ dd Z dd Z dd Zed eed eddfdZeddZddZddZdeddfdZededdfdZy) _ProgresszTrack aggregated and current progress. Args: total: Intended to track the total progress of an event. current: Intended to track the current progress of an event. default_factorytotalcurrentr Ncr|jj|jjur tdy)Nz?The `total` and `current` instances should be of the same class)rJr:rK ValueErrorr s r __post_init__z_Progress.__post_init__s/ ::  t||'='= =^_ _ >rc|jxjdz c_|jxjdz c_yN)rJr'rKr s rincrement_readyz_Progress.increment_readys, A arct|jts-td|jjj d|jxj dz c_|jxj dz c_y)N`z$` doesn't have a `started` attributerQ) isinstancerJr4 TypeErrorr:rr5rKr s rincrement_startedz_Progress.increment_starteds`$**o6a 4 4 = =>>bcd d a !rct|jts-td|jjj d|jxj dz c_|jxj dz c_y)NrTz&` doesn't have a `processed` attributerQ)rUrJrArVr:rrBrKr s rincrement_processedz_Progress.increment_processedsa$**&78a 4 4 = =>>def f ! !#rc|jxjdz c_|jxjdz c_yrP)rJr(rKr s rincrement_completedz_Progress.increment_completeds, ! !#r tracker_clskwargsc .||di||di|S)zZUtility function to easily create an instance from keyword arguments to both ``Tracker``s.)rJrKr$r$)rr\r]s r from_defaultsz_Progress.from_defaultss!.v. 8Mf8MNNrcl|jj|jjyr )rJrrKr s rrz_Progress.resets"  rc8|jjyr )rKrr s r reset_on_runz_Progress.reset_on_runs rc8|jjyr )rKr,r s rr,z_Progress.reset_on_restarts %%'rr-cp|jj||jj|yr )rJr0rKr/s rr0z_Progress.increment_bys& " !!!$rrc||jj|d|jj|dy)NrJrK)rJrrKrs rrz_Progress.load_state_dicts0 "":g#67 $$Z %:;rr)rrr r!rrArJr&r2rKrNrRrWrYr[r#typer1r_rrrbr,r0r"rr$rrrGrGs%*:K$LE !L&+r?s@rrhrhsv  M4 ####+c+$+4+9$9499rrhcJeZdZUdZeeZeed<eeZeed<y)_SchedulerProgressaTracks scheduler progress. These counters are local to a trainer rank. By default, they are not globally synced across all ranks. Args: total: Tracks the total scheduler progress. current: Tracks the current scheduler progress. rHrJrKN) rrr r!rr&rJr2rKr$rrrqrqs+%*:P$QE !Q&+z_OptimizerProgress.sI4K4KLb4crrHstepc4tjtSr )rGr_r4r$rrruz_OptimizerProgress.s9P9PQ`9ar zero_gradr Ncl|jj|jjyr )rvrrxr s rrz_OptimizerProgress.reset s   rcl|jj|jjyr )rvrbrxr s rrbz_OptimizerProgress.reset_on_runs"   ##%rcl|jj|jjyr )rvr,rxr s rr,z#_OptimizerProgress.reset_on_restarts" ""$ '')rrc||jj|d|jj|dy)Nrvrx)rvrrxrs rrz"_OptimizerProgress.load_state_dicts0 !!*V"45 &&z+'>?rr)rrr r!rrvrGr2rxrrrbr,r"rr$rrrsrssk,cdD)d 1abIyb &*@$@4@@rrsceZdZUdZeeZeed<ede fdZ e d dZ d dZ d d Ze d eddfd Zy) _OptimizationProgressz[Track optimization progress. Args: optimizer: Tracks optimizer progress. rH optimizerr cV|jjjjSr )rrvrJr(r s roptimizer_stepsz%_OptimizationProgress.optimizer_steps's~~""((222rNc8|jjyr )rrr s rrz_OptimizationProgress.reset+s rc8|jjyr )rrbr s rrbz"_OptimizationProgress.reset_on_run/s ##%rc8|jjyr )rr,r s rr,z&_OptimizationProgress.reset_on_restart2s '')rrc@|jj|dy)Nr)rrrs rrz%_OptimizationProgress.load_state_dict5s &&z+'>?rr)rrr r!rrsrr2propertyr1rrrrbr,r"rr$rrr~r~sw%*:L$MI!M 333&*@$@4@@rr~N) dataclassesrrrtyping_extensionsrrr&r4rArGrhrqrsr~r$rrrs10& "" "( ] D , <  > ;< ;< ;<| 9Y9 9D  T T  T @@ @> @M@ @r