o 4ѹg@sddlZddlZddlZddlZddlmZddlZddlm Z m Z dddZ dej fddZ dd d Zdd dZdddZddZddZddZddZdS)N) OrderedDict) DictConfig OmegaConfcCsZ|}d}tttdd|DD]\}}|dus"||r*|||7}q|S)NrcSsg|]}d|vr|qS)Znum_batches_tracked.0xrrU/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/site-packages/funasr/utils/misc.py sz.statistic_model_parameters..)Z state_dict enumeratesortedlistkeys startswithnumel)modelprefixZvar_dictrikeyrrr statistic_model_parameters s rcCs8dt|d|}tt|ddddk|S)Nz{:0zb}1)strformatnparrayr Zastype)rvec_dimZdtypebrrr int2vecs"rcstfdd|DS)Ncsg|] }tt|qSr)rintrrrr r szseq2arr..)rZ row_stack)seqrrr!r seq2arrsr#r c Cstj|ddd9}t}|D]'}|}||}|d|||dd}}|dkr3|d}|||<q|WdS1sDwYdS)Nrzutf-8encodingr r$)ioopenr readlinesstripfindsplit Zscp_pathZ value_typeZkv_sepfZret_dictZone_lineposrvaluerrr load_scp_as_dicts    $r3c Cstj|ddd;}g}|D]*}|}||}|d|||dd}}|dkr2|d}|||fq|WdS1sFwYdS)Nr%utf8r&r(r r$)r)r*r+r,r-r.appendr/rrr load_scp_as_list,s   $r6cCsT|D]#\}}t|tr#||vr#t|dkr|||<t|||q|||<qdS)Nr)items isinstancedictlen deep_update)originalupdaterr2rrr r;9s  r;c Kstj|ddddtj|ddd}tj||dtd|td||d d}|durStj|d }tj |rUt |tj|ddd dSdSdS) N output_dirz./T)exist_okz config.yaml)configr0zkwargs: zconfig.yaml is saved to: %s model_pathzconfiguration.json) osmakedirsgetpathjoinrsavelogginginfoexistsshutilcopy)kwargsZ yaml_filerAZ config_jsonrrr prepare_model_dirCs   rNcCs tj|}tj|\}}|S)u 从给定的文件路径中提取文件名（不包含路径和扩展名） :param file_path: 完整的文件路径 :return: 文件名（不含路径和扩展名） N)rBrEbasenamesplitext) file_pathZfilename_with_extensionfilename extensionrrr "extract_filename_without_extensionUs rTcCstj|st|ddStj|r$t|td|ddStj|rRzt|td|dWdStyQt |td|dYdSwdS)zVIntelligently removes files, empty directories, and non-empty directories recursively.z does not exist.NzFile z has been deleted.zEmpty directory zNon-empty directory z has been recursively deleted.) rBrErJprintisfileremoveisdirrmdirOSErrorrKrmtree)rErrr smart_removecs       r\)N)r)rr$)rBr)rKrH collectionsrZnumpyrZ omegaconfrrrZint32rr#r3r6r;rNrTr\rrrr s