idZddlmZmZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZddlmZGd d eZd e d e fd ZGddZy)z.Abstract base class used to build new loggers.)ABCabstractmethod) Namespace)wraps)AnyCallableOptionalUnion)Tensor)Module)rank_zero_onlyc NeZdZdZeedeefdZeedee e effdZ edeefdZ edeefdZ edefdZedd eeefd ee ddfd Zed e eeefefd ededdfdZddedeeddfdZddZdeddfdZy)Loggerz"Base class for experiment loggers.returncy)zReturn the experiment name.Nselfs t/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/lightning_fabric/loggers/logger.pynamez Logger.namecy)zReturn the experiment version.Nrrs rversionzLogger.version#rrcy)zReturn the root directory where all versions of an experiment get saved, or `None` if the logger does not save data locally.Nrrs rroot_dirzLogger.root_dir(rcy)z~Return directory the current version of the experiment gets saved, or `None` if the logger does not save data locally.Nrrs rlog_dirzLogger.log_dir.rrcy)zRReturn the default separator used by the logger to group the data into subfolders./rrs rgroup_separatorzLogger.group_separator4srNmetricsstepcy)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)rr#r$s r log_metricszLogger.log_metrics9s rparamsargskwargscy)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)rr'r(r)s rlog_hyperparamszLogger.log_hyperparamsDrrmodel input_arraycy)zRecord model graph. Args: model: the model with an implementation of ``forward``. input_array: input passes to `model.forward` Nr)rr,r-s r log_graphzLogger.log_graphOs rcy)zSave log data.Nrrs rsavez Logger.saveYrrstatusc$|jy)zDo any processing that is necessary to finalize an experiment. Args: status: Status that the experiment finished with (e.g. success, failed, aborted) N)r1)rr2s rfinalizezLogger.finalize\s  rN)rN)__name__ __module__ __qualname____doc__propertyrr strrr intrrrr"dictfloatr&rrr+r r r/r1r4rrrrrsf, *hsm**-%S/2--(3- #  4U #3 8C= TX   eDcNI,E&F s ^a fj   v HV4D PT strrfnrcbtdtdtttfffd }|S)zIReturns the real experiment on rank 0 and otherwise the _DummyExperiment.rrcNtjdkDr tS|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]``. r)r rank_DummyExperiment)rr?s r experimentz(rank_zero_experiment..experimentis%    "#% %$xr)rrr rrC)r?rDs` rrank_zero_experimentrEfs< 2Y  E#/?*?$@   rcXeZdZdZdededdfdZdedefdZd eddfd Z ded eddfd Z y) rCzDummy experiment.r(kwrNcyr5r)rr(rGs rnopz_DummyExperiment.nop~ r_c|jSr5)rI)rrKs r __getattr__z_DummyExperiment.__getattr__s xxridxc|Sr5r)rrNs r __getitem__z_DummyExperiment.__getitem__s rr)cyr5r)rr(r)s r __setitem__z_DummyExperiment.__setitem__rJr) r6r7r8r9rrIrrMr<rPrRrrrrCrC{s`  C D SXs'9    rrCN)r9abcrrargparser functoolsrtypingrrr r torchr torch.nnr $lightning_fabric.utilities.rank_zeror rrErCrrrrZsN5#11?HSHVX(*  r