o
    H!i	  ć                   @   s,   d dl mZ ddlmZ G dd deZdS )é   )Ś
ClsTraineré   )ŚMODELSc                   @   s   e Zd ZdZeZdd ZdS )ŚShiTuRecTrainerzShiTu Recognition Model Trainerc                 C   s&  | j jr| j | j j” | j jr| j | j j” | j jr$| j | j j” | j | j	j
d” | j jdur;| j | j j” | j jdkrI| j | j j” | j jdurW| j | j j” | j jdure| j | j j” | j jdurs| j | j j” | j jdur| j | j j” | j	jdur| j | j	j” dS dS )zupdate training configZShiTuRecDatasetNŚ )Ztrain_configZlog_intervalZ
pdx_configZupdate_log_intervalZeval_intervalZupdate_eval_intervalZsave_intervalZupdate_save_intervalZupdate_datasetZglobal_configZdataset_dirZnum_classesZupdate_num_classesZpretrain_weight_pathZupdate_pretrained_weightsZ
batch_sizeZupdate_batch_sizeZlearning_rateZupdate_learning_rateZepochs_itersZ_update_epochsZwarmup_stepsZupdate_warmup_epochsŚoutputZ_update_output_dir)Śself© r	   św/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddlex/modules/general_recognition/trainer.pyŚupdate_config   s4   ’’’zShiTuRecTrainer.update_configN)Ś__name__Ś
__module__Ś__qualname__Ś__doc__r   Śentitiesr   r	   r	   r	   r
   r      s    r   N)Zimage_classificationr   Z
model_listr   r   r	   r	   r	   r
   Ś<module>   s   