a  d@s~ddlZddlZddlZddlmZmZiZGdddZGdddZedd Z ed d Z d ej dfd dZ ddZ dS)N) get_dist_info master_onlyc@s6eZdZd ddZddZddZdd Zd d Zd S)AvgTimercCs*||_d|_d|_d|_d|_|dS)Nr)window current_time total_timecountavg_timestart)selfrr]/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/basicsr/utils/logger.py__init__ s zAvgTimer.__init__cCst|_|_dSN)time start_timeticr rrrr szAvgTimer.startcCsj|jd7_t|_|j|j|_|j|j7_|j|j|_|j|jkr\d|_d|_t|_dS)Nrr)r rZtocrrr r rrrrrrecords  zAvgTimer.recordcCs|jSr)rrrrrget_current_time&szAvgTimer.get_current_timecCs|jSr)r rrrr get_avg_time)szAvgTimer.get_avg_timeN)r)__name__ __module__ __qualname__rr rrrrrrrr s  rc@s.eZdZdZd ddZddZedd ZdS) MessageLoggeruMessage logger for printing. Args: opt (dict): Config. It contains the following keys: name (str): Exp name. logger (dict): Contains 'print_freq' (str) for logger interval. train (dict): Contains 'total_iter' (int) for total iters. use_tb_logger (bool): Use tensorboard logger. start_iter (int): Start iter. Default: 1. tb_logger (obj:`tb_logger`): Tensorboard logger. Default： None. rNcCsV|d|_|dd|_||_|dd|_|dd|_||_t|_t|_ dS)NnameloggerZ print_freqZtrainZ total_iter use_tb_logger) exp_nameinterval start_iter max_itersr tb_loggerrrget_root_loggerr)r optr"r$rrrr:s  zMessageLogger.__init__cCst|_dSr)rrrrrrreset_start_timeDszMessageLogger.reset_start_timecCsr|d}|d}|d}d|jddd|dd |d d }|D]}||d d 7}qH|d7}d|vr|d}|d}t|j} | ||jd} | |j|d} ttj t | d} |d| d7}|d|dd|dd7}| D]b\} }|| d|dd7}|j rd|jvr| drP|jd| ||q|j| ||q|j|dS)a\Format logging message. Args: log_vars (dict): It contains the following keys: epoch (int): Epoch number. iter (int): Current iter. lrs (list): List for learning rates. time (float): Iter time. data_time (float): Data time for each iter. epochiterlrs[Nz ..][epoch:Z3dz, iter:z8,dz, lr:(z.3e,z)] r data_timer)secondsz[eta: z, z time (data): z.3fz (z: z.4e debugZl_zlosses/)popr keysrrr"r#strdatetime timedeltaintitemsr startswithr$Z add_scalarrinfo)r Zlog_varsr(Z current_iterr*messagevZ iter_timer.r Z time_sec_avgZeta_secZeta_strkrrr__call__Gs.   &    zMessageLogger.__call__)rN)rrr__doc__rr'rr>rrrrr-s  rcCsddlm}||d}|S)Nr) SummaryWriter)log_dir)Ztorch.utils.tensorboardr@)rAr@r$rrrinit_tb_loggervs  rBcCsddl}t}|ddd}|ddd}|rP|}d}|d|d n|j}d }|j|||d ||d d |d|d|d dS)z.We now only use wandb to sync tensorboard log.rNrwandbproject resume_idZallowzResume wandb logger with id=.neverrT)idresumerconfigrDZsync_tensorboardzUse wandb logger with id=z ; project=)rCr%getwarningutilZ generate_idinitr:)r&rCrrDrEZwandb_idrIrrrinit_wandb_logger}s rOZbasicsrc Cst|}|tvr|Sd}t}|t|||d|_t\}}|dkr`| dnB|dur| |t |d}|t|| |||dt|<|S)aGet the root logger. The logger will be initialized if it has not been initialized. By default a StreamHandler will be added. If `log_file` is specified, a FileHandler will also be added. Args: logger_name (str): root logger name. Default: 'basicsr'. log_file (str | None): The log filename. If specified, a FileHandler will be added to the root logger. log_level (int): The root logger level. Note that only the process of rank 0 is affected, while other processes will set the level to "Error" and be silent most of the time. Returns: logging.Logger: The root logger. z&%(asctime)s %(levelname)s: %(message)sFrERRORNwT) logging getLoggerinitialized_logger StreamHandler setFormatter Formatter addHandler propagatersetLevel FileHandler) Z logger_nameZ log_levelZlog_filerZ format_strZstream_handlerZrank_Z file_handlerrrrr%s&        r%cCsBddl}ddl}ddlm}d}|d|d|jd|j7}|S)zPGet environment information. Currently, only log the software version. rN) __version__a] ____ _ _____ ____ / __ ) ____ _ _____ (_)_____/ ___/ / __ \ / __ |/ __ `// ___// // ___/\__ \ / /_/ / / /_/ // /_/ /(__ )/ // /__ ___/ // _, _/ /_____/ \__,_//____//_/ \___//____//_/ |_| ______ __ __ __ __ / ____/____ ____ ____/ / / / __ __ _____ / /__ / / / / __ / __ \ / __ \ / __ / / / / / / // ___// //_/ / / / /_/ // /_/ // /_/ // /_/ / / /___/ /_/ // /__ / /< /_/ \____/ \____/ \____/ \____/ /_____/\____/ \___//_/|_| (_) z! Version Information: BasicSR: z PyTorch: z TorchVision: )torch torchvisionZbasicsr.versionr])r^r_r]msgrrr get_env_infos  ra)r5rRrZ dist_utilrrrTrrrBrOINFOr%rarrrrs#I  *