
    9i
                     :    S SK r SSKJr  SSKJr   " S S\5      rg)    N   )BaseTrainer   )MODELSc                   0    \ rS rSrSr\rS rS\4S jr	Sr
g)
UadTrainer   zUad Model Trainerc                    U R                   R                  U R                  R                  S5        U R                  R
                  b/  U R                   R                  U R                  R
                  5        U R                  R                  (       aJ  U R                  R                  S:w  a/  U R                   R                  U R                  R                  SS9  ggg)zupdate training config
SegDatasetN T)is_backbone)	
pdx_configupdate_datasetglobal_configdataset_dirtrain_confignum_classesupdate_num_classespretrain_weight_pathupdate_pretrained_weights)selfs    i/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/modules/anomaly_detection/trainer.pyupdate_configUadTrainer.update_config   s    &&t'9'9'E'E|T((4OO..t/@/@/L/LM22!!66"<OO55!!66D 6  = 3    returnc                    SU R                  5       0n[        R                  R                  SS5        U R                  R
                  b  U R                  R
                  US'   U R                  R                  b  U R                  R                  US'   U R                  R                  b  U R                  R                  US'   U R                  R                  b3  U R                  R                  S:w  a  U R                  R                  US'   U R                  R                  b  U R                  R                  US	'   U R                  R                  (       a  U R                  R                  US
'   U R                  R                  (       a  SUS'   U R                  R                  US'   U R                  R                  SS5      US'   U R                  R                  SS5      US'   U$ )zkget key-value arguments of model training function

Returns:
    dict: the arguments of training function.
deviceFLAGS_npu_jit_compileN
batch_sizelearning_rateepochs_itersr   resume_pathsave_dir	log_itersTdo_evalsave_intervaldy2stFampOFF)
get_deviceosenvironpopr   r    r!   r"   r#   r   outputlog_intervaleval_intervalget)r   
train_argss     r   get_train_kwargsUadTrainer.get_train_kwargs(   s     12



.5''3'+'8'8'C'CJ|$**6*.*;*;*I*IJ'))5)-):):)G)GJ~&))5!!--3(,(9(9(E(EJ}%$$0%)%7%7%>%>Jz"))&*&7&7&D&DJ{#**$(Jy!*.*;*;*I*IJ'"//33GUC
7 --11%?
5r    N)__name__
__module____qualname____firstlineno____doc__r   entitiesr   dictr4   __static_attributes__r6   r   r   r   r      s    H$ r   r   )r,   baser   
model_listr   r   r6   r   r   <module>rA      s     
  0 0r   