
    |-j                     N    d dl Z d dlmZ ddlmZ ddlmZ  G d de          ZdS )	    N)Path   )BaseTrainer   )MODELSc                   4    e Zd ZdZeZdefdZd Zde	fdZ
dS )TextRecTrainerzText Recognition Model Trainersrc_label_dict_pathc                     t          | j        j                                      d          }t	          j        ||           dS )zzdump label dict config

        Args:
            src_label_dict_path (str): path to label dict file to be saved.
        zlabel_dict.txtN)r   global_configoutputjoinpathshutilcopyfile)selfr
   dst_label_dict_paths      h/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/modules/text_recognition/trainer.pydump_label_dictzTextRecTrainer.dump_label_dict   sA     #4#5#<==FFGWXX+-@AAAAA    c                 <   | j         j        r$| j                            | j         j                   | j         j        r$| j                            | j         j                   | j         j        r$| j                            | j         j                   | j        d         dk    r&| j        	                    | j        j
        d           nZd| j        d         v r&| j        	                    | j        j
        d           n%| j        	                    | j        j
        d           t          | j        j
                                      d          }|                                r/| j                            |           |                     |           | j         j        r$| j                            | j         j                   | j        d         dk    rH| j         j        ;| j         j        r/| j                            | j         j        | j         j                   n0| j         j        $| j                            | j         j                   | j         j        $| j                            | j         j                   | j         j        $| j                            | j         j                   | j         j        4| j         j        d	k    r$| j                            | j         j                   | j        j        &| j                            | j        j                   dS dS )
zupdate training configmodelLaTeX_OCR_recLaTeXOCRDataSetzPP-OCRv3SimpleDataSetMSTextRecDatasetzdict.txtN )train_configlog_interval
pdx_configupdate_log_intervaleval_interval_update_eval_interval_by_epochsave_intervalupdate_save_intervalr   update_datasetdataset_dirr   r   existsupdate_label_dict_pathr   pretrain_weight_pathupdate_pretrained_weightsbatch_size_trainbatch_size_valupdate_batch_size_pair
batch_sizeupdate_batch_sizelearning_rateupdate_learning_rateepochs_iters_update_epochsresume_path_update_checkpointsr   _update_output_dir)r   label_dict_paths     r   update_configzTextRecTrainer.update_config%   s   ) 	PO//0A0NOOO* 	O::!/   * 	RO001B1PQQQg&/99O**".0A    4-g666O**".    O**".0B   t1=>>GG
SS!!## 	2O22?CCC  1111 	O55!6   g&/99!2>%4 ? 66%68I8X    +711$2C2NOOO*6O001B1PQQQ)5O**4+<+IJJJ)5!-33O//0A0MNNN$0O..t/A/HIIIII 10r   returnc                     |                                  | j                            dd          | j                            dd          dS )zget key-value arguments of model training function

        Returns:
            dict: the arguments of training function.
        dy2stFampOFF)devicer;   r<   )
get_devicer   get)r   s    r   get_train_kwargszTextRecTrainer.get_train_kwargs_   sL     oo''&**7E::$((66
 
 	
r   N)__name__
__module____qualname____doc__r   entitiesstrr   r8   dictrA    r   r   r	   r	      sm        ((HB3 B B B B8J 8J 8Jt

$ 

 

 

 

 

 

r   r	   )r   pathlibr   baser   
model_listr   r	   rI   r   r   <module>rM      s                       R
 R
 R
 R
 R
[ R
 R
 R
 R
 R
r   