a þdàã@sždZddlmZmZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZGd d „d eƒZe e d œd d „ZGdd„dƒZdS)z.Abstract base class used to build new loggers.é)ÚABCÚabstractmethod)Ú Namespace)Úwraps)ÚAnyÚCallableÚDictÚOptionalÚUnion)ÚTensor)ÚModule)Úrank_zero_onlyc@s eZdZdZeeeedœdd„ƒƒZeeee e efdœdd„ƒƒZ eeedœdd„ƒZ eeedœd d „ƒZ eedœd d „ƒZedeeefee d dœdd„ƒZee eeefefeed dœdd„ƒZdeeed dœdd„Zd dœdd„Zed dœdd„Zd S)ÚLoggerz"Base class for experiment loggers.©ÚreturncCsdS)zReturn the experiment name.N©©Úselfrrúh/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/lightning_fabric/loggers/logger.pyÚnamesz Logger.namecCsdS)zReturn the experiment version.NrrrrrÚversion#szLogger.versioncCsdS)z„Return the root directory where all versions of an experiment get saved, or `None` if the logger does not save data locally.NrrrrrÚroot_dir(szLogger.root_dircCsdS)z~Return directory the current version of the experiment gets saved, or `None` if the logger does not save data locally.NrrrrrÚlog_dir.szLogger.log_dircCsdS)zRReturn the default separator used by the logger to group the data into subfolders.ú/rrrrrÚgroup_separator4szLogger.group_separatorN)ÚmetricsÚsteprcCsdS)aRecords metrics. This method logs metrics as soon as it received them. Args: metrics: Dictionary with metric names as keys and measured quantities as values step: Step number at which the metrics should be recorded Nr)rrrrrrÚ log_metrics9szLogger.log_metrics)ÚparamsÚargsÚkwargsrcOsdS)a>Record hyperparameters. Args: params: :class:`~argparse.Namespace` or `Dict` containing the hyperparameters args: Optional positional arguments, depends on the specific logger being used kwargs: Optional keyword arguments, depends on the specific logger being used Nr)rrrr rrrÚlog_hyperparamsCszLogger.log_hyperparams)ÚmodelÚ input_arrayrcCsdS)z¨Record model graph. Args: model: the model with an implementation of ``forward``. input_array: input passes to `model.forward` Nr)rr"r#rrrÚ log_graphMszLogger.log_graphcCsdS)zSave log data.NrrrrrÚsaveVsz Logger.save)ÚstatusrcCs | ¡dS)z³Do any processing that is necessary to finalize an experiment. Args: status: Status that the experiment finished with (e.g. success, failed, aborted) N)r%)rr&rrrÚfinalizeYszLogger.finalize)N)N)Ú__name__Ú __module__Ú __qualname__Ú__doc__Úpropertyrr Ústrrr ÚintrrrrrÚfloatrrrr!r r r$r%r'rrrrrs(" &  r)Úfnrcs&tˆƒtttfdœ‡fdd„ ƒ}|S)zIReturns the real experiment on rank 0 and otherwise the _DummyExperiment.rcs$ttdœ‡‡fdd„ ƒ}|ƒp"tƒS)aÈ Note: ``self`` is a custom logger instance. The loggers typically wrap an ``experiment`` method with a ``@rank_zero_experiment`` decorator. ``Union[Any, _DummyExperiment]`` is used because the wrapped hooks have several return types that are specific to the custom logger. The return type here can be considered as ``Union[return type of logger.experiment, _DummyExperiment]``. rcsˆˆƒS©Nrr)r0rrrÚget_experimentqsz@rank_zero_experiment..experiment..get_experiment)r rÚ_DummyExperiment)rr2©r0rrÚ experimentes z(rank_zero_experiment..experiment)rr rr3)r0r5rr4rÚrank_zero_experimentbsr6c@sTeZdZdZeeddœdd„Zeedœdd„Zedd œd d „Z eedd œd d„Z dS)r3zDummy experiment.N)rÚkwrcOsdSr1r)rrr7rrrÚnop}sz_DummyExperiment.nop)Ú_rcCs|jSr1)r8)rr9rrrÚ __getattr__€sz_DummyExperiment.__getattr__)ÚidxrcCs|Sr1r)rr;rrrÚ __getitem__ƒsz_DummyExperiment.__getitem__)rr rcOsdSr1r)rrr rrrÚ __setitem__‡sz_DummyExperiment.__setitem__) r(r)r*r+rr8rr:r.r<r=rrrrr3zs r3N)r+ÚabcrrÚargparserÚ functoolsrÚtypingrrrr r Ztorchr Ztorch.nnr Z$lightning_fabric.utilities.rank_zeror rr6r3rrrrÚs     G