
    9i~
                     2    S SK Jr  SSKJr   " S S\5      rg)   )BaseTrainer   )MODELSc                   0    \ rS rSrSr\rS rS\4S jr	Sr
g)TableRecTrainer   zTable Recognition Model Trainerc                    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                  R                  U R                  R                  S5        U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R                   R                  b/  U R                  R                  U R                   R                  5        U R                   R                  b/  U R                  R!                  U R                   R                  5        U R                   R"                  b/  U R                  R%                  U R                   R"                  5        U R                   R&                  bI  U R                   R&                  S:w  a/  U R                  R)                  U R                   R&                  5        U R                  R*                  b0  U R                  R-                  U R                  R*                  5        gg)zupdate training configPubTabTableRecDatasetN )train_configlog_interval
pdx_configupdate_log_intervaleval_interval_update_eval_interval_by_epochsave_intervalupdate_save_intervalupdate_datasetglobal_configdataset_dirpretrain_weight_pathupdate_pretrained_weights
batch_sizeupdate_batch_sizelearning_rateupdate_learning_rateepochs_iters_update_epochsresume_path_update_checkpointsoutput_update_output_dirselfs    i/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/modules/table_recognition/trainer.pyupdate_configTableRecTrainer.update_config   s   ))OO//0A0A0N0NO**OO::!!// **OO001B1B1P1PQ&&**,C	
 11OO55!!66 ''3OO--d.?.?.J.JK**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 1    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   getr#   s    r%   get_train_kwargs TableRecTrainer.get_train_kwargs9   sD     oo'&&**7E:$$((6
 	
r(    N)__name__
__module____qualname____firstlineno____doc__r   entitiesr&   dictr1   __static_attributes__r3   r(   r%   r   r      s    )HJ@

$ 

r(   r   N)baser   
model_listr   r   r3   r(   r%   <module>r>      s      /
k /
r(   