o 1Îj] ã@s,ddlmZddlmZGdd„deƒZdS)é)Ú BaseTraineré)ÚMODELSc@s*eZdZdZeZdd„Zdefdd„ZdS)ÚTextDetTrainerzText Detection Model TrainercCs|jjr |j |jj¡|jjr|j |jj¡|jjr$|j |jj¡|j |j j d¡|jj r9|j  |jj ¡|jj durG|j |jj ¡|jjdurU|j |jj¡|jjdurc|j |jj¡|jjdurw|jjdkrw|j |jj¡|j jdur‡|j |j j¡dSdS)zupdate training configZTextDetDatasetNÚ)Ú train_configZ log_intervalZ pdx_configZupdate_log_intervalZ eval_intervalZ_update_eval_interval_by_epochZ save_intervalZupdate_save_intervalZupdate_datasetZ global_configZ dataset_dirZpretrain_weight_pathZupdate_pretrained_weightsZ batch_sizeZupdate_batch_sizeZ learning_rateZupdate_learning_rateZ epochs_itersZ_update_epochsZ resume_pathZ_update_checkpointsÚoutputZ_update_output_dir©Úself©r úg/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/text_detection/trainer.pyÚ update_configs2ÿÿ      ÿzTextDetTrainer.update_configÚreturncCs$| ¡|j dd¡|j dd¡dœS)zƒget key-value arguments of model training function Returns: dict: the arguments of training function. Údy2stFÚampZOFF)Zdevicerr)Z get_devicerÚgetr r r r Úget_train_kwargs7s  ýzTextDetTrainer.get_train_kwargsN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚentitiesr Údictrr r r r rs rN)ÚbaserZ model_listrrr r r r Ús