
    |-j	                     :    d dl mZ ddlmZ  G d de          ZdS )   )
ClsTrainer   )MODELSc                       e Zd ZdZeZd ZdS )ShiTuRecTrainerzShiTu Recognition Model Trainerc                    | j         j        r$| j                            | j         j                   | j         j        r$| j                            | j         j                   | j         j        r$| j                            | j         j                   | j                            | j	        j
        d           | j         j        $| j                            | j         j                   | j         j        dk    r$| j                            | j         j                   | j         j        $| j                            | j         j                   | j         j        $| j                            | j         j                   | j         j        $| j                            | j         j                   | j         j        $| j                            | j         j                   | j	        j        &| j                            | j	        j                   dS dS )zupdate training configShiTuRecDatasetN )train_configlog_interval
pdx_configupdate_log_intervaleval_intervalupdate_eval_intervalsave_intervalupdate_save_intervalupdate_datasetglobal_configdataset_dirnum_classesupdate_num_classespretrain_weight_pathupdate_pretrained_weights
batch_sizeupdate_batch_sizelearning_rateupdate_learning_rateepochs_iters_update_epochswarmup_stepsupdate_warmup_epochsoutput_update_output_dir)selfs    k/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/modules/general_recognition/trainer.pyupdate_configzShiTuRecTrainer.update_config   s   ) 	PO//0A0NOOO* 	RO001B1PQQQ* 	RO001B1PQQQ&&*,=	
 	
 	
 (4O..t/@/LMMM1R77O55!6   '3O--d.?.JKKK*6O001B1PQQQ)5O**4+<+IJJJ)5O001B1OPPP$0O..t/A/HIIIII 10    N)__name__
__module____qualname____doc__r   entitiesr&    r'   r%   r   r      s3        ))HJ J J J Jr'   r   N)image_classificationr   
model_listr   r   r-   r'   r%   <module>r0      si    . - - - - -      !J !J !J !J !Jj !J !J !J !J !Jr'   