a d@s ddlmZddlmZmZmZmZmZmZddl Z ddl m Z eeee efefee efdddZee efee efddd Zdeeefe e ee efd d dZee efee efdddZee ee effe e ee ee effdddZdS)) Namespace)AnyDictMappingMutableMappingOptionalUnionN)Tensor)paramsreturncCs"t|trt|}|duri}|S)zEnsure parameters are a dict or convert to dict if necessary. Args: params: Target to be converted to a dictionary Returns: params as a dictionary N) isinstancervarsr rj/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/lightning_fabric/utilities/logger.py_convert_paramss rcs&ttdddfdd|DS)zSanitize callable params dict, e.g. ``{'a': } -> {'a': 'function_****'}``. Args: params: Dictionary containing the hyperparameters Returns: dictionary with all callables sanitized )valr cSsHt|rDz|}t|r |jWS|WStyBt|ddYS0|S)N__name__)callabler Exceptiongetattr)rZ_valrrr_sanitize_callable3s z5_sanitize_callable_params.._sanitize_callablecsi|]\}}||qSrr).0keyrrrr @z-_sanitize_callable_params..)ritemsrrrr_sanitize_callable_params)s  r/)r delimiter parent_keyr cCsri}|D]`\}}|r(||t|nt|}t|trBt|}t|trdi|t|||d}q |||<q |S)aFlatten hierarchical dict, e.g. ``{'a': {'b': 'c'}} -> {'a/b': 'c'}``. Args: params: Dictionary containing the hyperparameters delimiter: Delimiter to express the hierarchy. Defaults to ``'/'``. Returns: Flattened dict. Examples: >>> _flatten_dict({'a': {'b': 'c'}}) {'a/b': 'c'} >>> _flatten_dict({'a': {'b': 123}}) {'a/b': 123} >>> _flatten_dict({5: {'a': 123}}) {'5/a': 123} )r"r!)rstrr rr r _flatten_dict)r r!r"resultkvZnew_keyrrrr$Cs   r$cCsh|D]Z}t||tjtjtjfr8||||<qt||tt t t t fvrt ||||<q|S)aReturns params with non-primitvies converted to strings for logging. >>> import torch >>> params = {"float": 0.3, ... "int": 1, ... "string": "abc", ... "bool": True, ... "list": [1, 2, 3], ... "namespace": Namespace(foo=3), ... "layer": torch.nn.BatchNorm1d} >>> import pprint >>> pprint.pprint(_sanitize_params(params)) # doctest: +NORMALIZE_WHITESPACE {'bool': True, 'float': 0.3, 'int': 1, 'layer': "", 'list': '[1, 2, 3]', 'namespace': 'Namespace(foo=3)', 'string': 'abc'} ) keysr npZbool_integerZfloatingitemtypeboolintfloatr#r )r r&rrr_sanitize_paramsas  r0)metricsprefix separatorr cs rfdd|D}|S)akInsert prefix before each key in a dict, separated by the separator. Args: metrics: Dictionary with metric names as keys and measured quantities as values prefix: Prefix to insert before each key separator: Separates prefix and original key name Returns: Dictionary with prefix and separator inserted before each key cs"i|]\}}||qSrr)rr&r'r2r3rrrrz_add_prefix..)r)r1r2r3rr4r _add_prefixs r5)rr )argparsertypingrrrrrrnumpyr)Ztorchr r#rrr$r0r/r5rrrrs   , &