o H!i @s4ddlZddlmZddlmZGdddeZdS)N) BaseTrainer)MODELSc@s*eZdZdZeZddZdefddZdS) UadTrainerzUad Model TrainercCsb|j|jjd|jjdur|j|jj|jjr-|jjdkr/|jj|jjdddSdSdS)zupdate training configZ SegDatasetNT)Z is_backbone) Z pdx_configZupdate_dataset global_configZ dataset_dir train_configZ num_classesZupdate_num_classesZpretrain_weight_pathZupdate_pretrained_weights)selfr u/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddlex/modules/anomaly_detection/trainer.py update_configs   zUadTrainer.update_configreturncCsd|i}tjdd|jjdur|jj|d<|jjdur%|jj|d<|jjdur1|jj|d<|jjdurC|jjdkrC|jj|d<|j j durO|j j |d <|jj rY|jj |d <|jj rgd |d <|jj |d <|j dd|d<|j dd|d<|S)zget key-value arguments of model training function Returns: dict: the arguments of training function. ZdeviceZFLAGS_npu_jit_compileN batch_size learning_rate epochs_itersr resume_pathsave_dirZ log_itersTZdo_evalZ save_intervalZdy2stFampZOFF)Z get_deviceosenvironpopr rrrrroutputZ log_intervalZ eval_intervalget)r Z train_argsr r r get_train_kwargs(s*              zUadTrainer.get_train_kwargsN) __name__ __module__ __qualname____doc__rentitiesr dictrr r r r rs  r)rbaserZ model_listrrr r r r s