o zi@sdZddlmZmZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZGd d d eZd e d e fd dZGdddZdS)z.Abstract base class used to build new loggers.)ABCabstractmethod) Namespace)wraps)AnyCallableDictOptionalUnion)Tensor)Module)rank_zero_onlyc @seZdZdZeedeefddZeedee e effddZ edeefddZ edeefd d Z edefd d Zed deeefdee dd fddZede eeefefdededd fddZd dedeedd fddZd!ddZdedd fddZd S)"Loggerz"Base class for experiment loggers.returncCdS)zReturn the experiment name.NselfrrS/home/ubuntu/.local/lib/python3.10/site-packages/lightning_fabric/loggers/logger.pynamez Logger.namecCr)zReturn the experiment version.Nrrrrrversion#rzLogger.versioncCr)zReturn the root directory where all versions of an experiment get saved, or `None` if the logger does not save data locally.Nrrrrrroot_dir(zLogger.root_dircCr)z~Return directory the current version of the experiment gets saved, or `None` if the logger does not save data locally.Nrrrrrlog_dir.rzLogger.log_dircCr)zRReturn the default separator used by the logger to group the data into subfolders./rrrrrgroup_separator4szLogger.group_separatorNmetricsstepcCr)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_metrics9s zLogger.log_metricsparamsargskwargscOr)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"rrrlog_hyperparamsDrzLogger.log_hyperparamsmodel input_arraycCr)zRecord model graph. Args: model: the model with an implementation of ``forward``. input_array: input passes to `model.forward` Nr)rr$r%rrr log_graphOszLogger.log_graphcCr)zSave log data.NrrrrrsaveYrz Logger.savestatuscCs |dS)zDo 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(rrrfinalize\s zLogger.finalizeN)rN)__name__ __module__ __qualname____doc__propertyrr strrr intrrrrrfloatrrrr#r r r&r'r)rrrrrs*& , rfnrcs*tdtdtttfffdd }|S)zIReturns the real experiment on rank 0 and otherwise the _DummyExperiment.rrcstjdkrtS|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_DummyExperimentrr3rr experimentis z(rank_zero_experiment..experiment)rrr rr5)r3r7rr6rrank_zero_experimentfs r8c@s`eZdZdZdededdfddZdedefd d Zd eddfd d Z dededdfddZ dS)r5zDummy experiment.r!kwrNcOdSr*r)rr!r9rrrnop~z_DummyExperiment.nop_cCs|jSr*)r;)rr=rrr __getattr__sz_DummyExperiment.__getattr__idxcCs|Sr*r)rr?rrr __getitem__sz_DummyExperiment.__getitem__r"cOr:r*r)rr!r"rrr __setitem__r<z_DummyExperiment.__setitem__) r+r,r-r.rr;rr>r1r@rArrrrr5{s r5N)r.abcrrargparser functoolsrtypingrrrr r torchr torch.nnr $lightning_fabric.utilities.rank_zeror rr8r5rrrrs      K