|-j] :ddlmZddlmZGddeZdS)) BaseTrainer)MODELSc(eZdZdZeZdZdefdZdS)TextDetTrainerzText Detection Model Trainerc|jjr$|j|jj|jjr$|j|jj|jjr$|j|jj|j|j j d|jj r$|j |jj |jj $|j|jj |jj$|j|jj|jj$|j|jj|jj4|jjdkr$|j|jj|j j&|j|j jdSdS)zupdate training configTextDetDatasetN) train_config log_interval pdx_configupdate_log_interval eval_interval_update_eval_interval_by_epoch save_intervalupdate_save_intervalupdate_dataset global_config dataset_dirpretrain_weight_pathupdate_pretrained_weights batch_sizeupdate_batch_size learning_rateupdate_learning_rate epochs_iters_update_epochs resume_path_update_checkpointsoutput_update_output_dirselfs f/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/modules/text_detection/trainer.py update_configzTextDetTrainer.update_configs   ) P O / /0A0N O O O   *  O : :!/      * R O 0 01B1P Q Q Q &&t'9'EGWXXX   1  O 5 5!6      ' 3 O - -d.?.J K K K   * 6 O 0 01B1P Q Q Q   ) 5 O * *4+<+I J J J   ) 5!-33 O / /0A0M N N N   $ 0 O . .t/A/H I I I I I 1 0returnc||jdd|jdddS)zget key-value arguments of model training function Returns: dict: the arguments of training function. dy2stFampOFF)devicer)r*) get_devicer getr"s r$get_train_kwargszTextDetTrainer.get_train_kwargs7sLoo''&**7E::$((66   r&N) __name__ __module__ __qualname____doc__rentitiesr%dictr/r&r$rrsL&&HJJJ<  $       r&rN)baser model_listrrr6r&r$r9s_ - - - - - [- - - - - r&