i'dZddlmZddlmZmZddlZddlm Z ddl m Z m Z m Z ddlmZddlmZmZmZmZmZmZdd lmZdd lmZmZd d d eeeefdeeeefdeeeefdeeeefdeeefdeeefdeeeeeee fdeddfdZ!deeeefdedeeeffdZ"d d deeeefddfdZ#ddZ$deeee fdefdZ%y) z.Houses the methods used to set up the Trainer.) timedelta)OptionalUnionN)PossibleUserWarning)CUDAAcceleratorMPSAcceleratorXLAAccelerator) DummyLogger)AdvancedProfilerPassThroughProfilerProfilerPyTorchProfilerSimpleProfiler XLAProfiler)MisconfigurationException)rank_zero_inforank_zero_warntrainer pl.Trainerlimit_train_batcheslimit_val_batcheslimit_test_batcheslimit_predict_batches fast_dev_runoverfit_batchesval_check_intervalnum_sanity_val_stepsreturnc Dt|tr|dkrtd|d||_|dk(rd|_t |d|_|dkD} |rt|} | s| |_| |_| |_| |_ | |jj_ d|_ d|j_d|_d|_d|_|j$r t'gng|_t)d| d n| s"t |d |_t |d |_t |d |_t |d |_ |dk(r t+dn||_ d|_t|t,t.t0frt3||_nt |d|_| r||_||_yy)Nrz fast_dev_run=z1 is not a valid configuration. It should be >= 0.Tr?zBRunning in `fast_dev_run` mode: will run the requested loop using z4 batch(es). Logging and checkpointing is suppressed.rrrrinfr) isinstanceintrr_determine_batch_limitsrrrrrfit_loop epoch_loop max_stepsr max_epochsr_val_check_time_intervalcheck_val_every_n_epochloggersr rfloatstrdictr_parse_time_interval_seconds) rrrrrrrrroverfit_batches_enabled num_batchess t/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pytorch_lightning/trainer/setup.py_init_debugging_flagsr5#s,$,*:'L++\ ]  (Gq#5oGXYG-1,' &*5G '(3G %%0"(3%0;##-'($&'#%("+/(*+'-4__;=/"PQ\P]^7 7 '*ABUWl*mG '(?@QSf(gG %%<=OQe%f"(?@UWn(o%7Kr7QuU|Wk$,0( (3i*@ A/KL^/_G ,)@ASUi)jG &&5#$3!batchesnamec\|yt|tr)|dk(r$|dk(rd}n |dk(rd}nd}td|d |n9t|tr)|dk(r$|dk(rd }n |dk(rd }nd }td|d |dd|cxkrdkr|S|dkDr|dzdk(r t|St d|d|d)Nr!r rz1 batch per epoch will be used.rz&validation will run after every batch.z1 batch will be used.z `Trainer(z=1)` was configured so z+100% of the batches per epoch will be used.z5validation will run at the end of the training epoch.z!100% of the batches will be used.z=1.0)` was configured so .rzYou have passed invalid value z for z', it has to be in [0.0, 1.0] or an int.)r$r%rr.r)r7r8messages r4r&r&ds'3GqL ( (7G ) )>G-G4&(?yIJ GU #3 ( (CG ) )MG9G4&(A'!LMGq{w})7| # ( tf># 9 $+**.y~~/?*@)AC #8,!#8#6#8Gr6ctjrd}d}ntjrd}d}nd}d}t|jttf}t d||d|t|jt r |jnd}t d t jd |d tjrt|jtr.tjr+t|jtstd t t jr't|jt s tdyyy)NTz (cuda)z (mps)FzGPU available: z, used: rzTPU available: z , using: z TPU coreszQGPU available but not used. You can set it by doing `Trainer(accelerator='gpu')`.)categoryzQTPU available but not used. You can set it by doing `Trainer(accelerator='tpu')`.) r is_availablerr$ acceleratorrr num_devicesrr)r gpu_availablegpu_typegpu_used num_tpu_coress r4_log_device_inforRs##%   $ $ &  '--/PQH_]OH:XhZPQ+5g6I6I>+ZG''`aM_^%@%@%B$C9]O[efg $$&7..@  & & (7..? _( ""$Z8K8K^-\jk.]$r6valuec t|tr|jSt|trtd i|}|jSt|tr|j d}t |dk7rtd|d|\}}}} t|}t|}t|} t|} t||| | }|jStdt|#t$rtd|dwxYw) a]Convert a time interval into seconds. This helper parses different representations of a time interval and normalizes them into a float number of seconds. Supported input formats: * `timedelta`: The total seconds are returned directly. * `dict`: A dictionary of keyword arguments accepted by `datetime.timedelta`, e.g. `{"days": 1, "hours": 2}`. * `str`: A string in the format `"DD:HH:MM:SS"`, where each component must be an integer. Args: value (Union[str, timedelta, dict]): The time interval to parse. Returns: float: The duration represented by `value` in seconds. Raises: MisconfigurationException: If the input type is unsupported, the string format is invalid, or any string component is not an integer. Examples: >>> _parse_time_interval_seconds("01:02:03:04") 93784.0 >>> _parse_time_interval_seconds({"hours": 2, "minutes": 30}) 9000.0 >>> from datetime import timedelta >>> _parse_time_interval_seconds(timedelta(days=1, seconds=30)) 86430.0 :z.Invalid time format for `val_check_interval`: z. Expected 'DD:HH:MM:SS'.z6Non-integer component in `val_check_interval` string: z. Use 'DD:HH:MM:SS'.)dayshoursminutessecondsz+Unsupported type for `val_check_interval`: ) r$r total_secondsr0r/splitlenrr% ValueErrortype) rStdpartsdhmsrWrXrYrZs r4r1r1s F%#""$$%   !!% C  u:?+@ Ibc  1a q6DFE!fG!fG Dw P!! #&QRVW\R]Q`$a bb +H Qef  s ,C44D )rrrN)&__doc__datetimertypingrrpytorch_lightningpl#lightning_fabric.utilities.warningsrpytorch_lightning.acceleratorsrrr pytorch_lightning.loggers.loggerr pytorch_lightning.profilersr r r rrr&pytorch_lightning.utilities.exceptionsr%pytorch_lightning.utilities.rank_zerorrr%r.boolr/r0r5r&rGrRr1r[r6r4rss5"CZZ8MP>4 >4!%U "34>4 c5j 12>4!sEz!23 >4 $E#u*$56 >4 T " >43:&>4!sE3 4'G!HI>4>4 >4BXeCJ.?%@PUVY[`V`PaB9L9HU8S==Q4R9W[9&l@;cc9d.B(C;c;cr6