
    9i                     F    S SK r S SKJr  SSKJr  SSKJr   " S S\5      rg)	    N)Path   )BaseTrainer   )MODELSc                   >    \ rS rSrSr\rS\4S jrS r	S\
4S jrSrg	)
TextRecTrainer   zText Recognition Model Trainersrc_label_dict_pathc                     [        U R                  R                  5      R                  S5      n[        R
                  " X5        g)zbdump 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/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/modules/text_recognition/trainer.pydump_label_dictTextRecTrainer.dump_label_dict   s4     #4#5#5#<#<=FFGWX+A    c                    U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R                  S   S:X  a1  U R                  R                  U R                  R                  S5        OtSU R                  S   ;   a1  U R                  R                  U R                  R                  S5        O0U R                  R                  U R                  R                  S5        [        U R                  R                  5      R                  S5      nUR                  5       (       a,  U R                  R                  U5        U R                  U5        U R                   R                   (       a/  U R                  R#                  U R                   R                   5        U R                  S   S:X  aw  U R                   R$                  b_  U R                   R&                  (       aD  U R                  R)                  U R                   R$                  U R                   R&                  5        OFU R                   R*                  b/  U R                  R-                  U R                   R*                  5        U R                   R.                  b/  U R                  R1                  U R                   R.                  5        U R                   R2                  b/  U R                  R5                  U R                   R2                  5        U R                   R6                  bI  U R                   R6                  S	:w  a/  U R                  R9                  U R                   R6                  5        U R                  R:                  b0  U R                  R=                  U R                  R:                  5        gg)
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_configTextRecTrainer.update_config%   s   ))OO//0A0A0N0NO**OO::!!// **OO001B1B1P1PQg&/9OO**""..0A 4--g66OO**"".. OO**""..0B t11==>GG
S!!##OO22?C  111OO55!!66 g&/9!!22>%%4466%%668I8I8X8X   ++711$2C2C2N2NO**6OO001B1B1P1PQ))5OO**4+<+<+I+IJ))5!!--3OO//0A0A0M0MN$$0OO..t/A/A/H/HI 1r   returnc                     U R                  5       U R                  R                  SS5      U R                  R                  SS5      S.$ )zkget 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_kwargsTextRecTrainer.get_train_kwargs_   sD     oo'&&**7E:$$((6
 	
r    N)__name__
__module____qualname____firstlineno____doc__r   entitiesstrr   r:   dictrD   __static_attributes__rF   r   r   r	   r	      s+    (HB3 B8Jt

$ 

r   r	   )r   pathlibr   baser   
model_listr   r	   rF   r   r   <module>rS      s!        R
[ R
r   