o %ݫi @s@dZddlZddlZddlmZeeZ     dddZdS) zK Training utilities for pretrained models Authors * Artem Ploujnikov 2021 N) get_logger pretrainer checkpointercsJtfdd||fDrtd|d|d|}|}|j|||d}|rtd|jt|j} t|j } | | @} | D]<} |j | } t j | sZtd| d |j | }t j |}t j |spt |t j |r{t |t| |qEd }|Std |||}d d d|D}td|d}|S)ao Saves the necessary files for the pretrained model from the best checkpoint found. The goal of this function is to export the model for a Pretrainer Arguments --------- hparams: dict the hyperparameter file min_key: str Key to use for finding best checkpoint (lower is better). By default, passed to ``self.checkpointer.recover_if_possible()``. max_key: str Key to use for finding best checkpoint (higher is better). By default, passed to ``self.checkpointer.recover_if_possible()``. ckpt_predicate: callable a filter predicate to locate checkpoints pretrainer_key: str the key under which the pretrainer is stored checkpointer_key: str the key under which the checkpointer is stored Returns ------- saved: bool Whether the save was successful c3s|]}|vVqdS)N).0keyhparamsrP/home/ubuntu/.local/lib/python3.10/site-packages/speechbrain/utils/pretrained.py 3sz&save_for_pretrained..z.Incompatible hparams: a checkpointer with key zand a pretrainer with key z are required)min_keymax_keyckpt_predicatez)Saving checkpoint '%s' a pretrained modelzFile z! does not exist in the checkpointTzCUnable to find a matching checkpoint for min_key = %s, max_key = %s css"|] }|jd|jVqdS)z: N)pathmeta)r checkpointrrr r Ys  zAvailable checkpoints: %sF)any ValueErrorfind_checkpointloggerinforset loadableskeys paramfilesosexistspathsdirnamemakedirsremoveshutilcopyfilelist_checkpointsjoin)r r r rpretrainer_keycheckpointer_keyrrrpretrainer_keyscheckpointer_keys keys_to_saver source_path target_pathrsaved checkpointscheckpoints_strrrr save_for_pretrainedsZ#           r0)NNNrr)__doc__rr"speechbrain.utils.loggerr__name__rr0rrrr s