
    {-j
                     B    d dl Z ddlmZ ddlmZ  G d de          ZdS )    N   )BaseTrainer   )MODELSc                   (    e Zd ZdZeZd ZdefdZdS )
UadTrainerzUad Model Trainerc                 <   | j                             | j        j        d           | j        j        $| j                             | j        j                   | j        j        r8| j        j        dk    r*| j                             | j        j        d           dS dS dS )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/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/modules/anomaly_detection/trainer.pyupdate_configzUadTrainer.update_config   s    &&t'9'E|TTT(4O..t/@/LMMM2	!6"<<O55!6D 6     		 	<<    returnc                    d|                                  i}t          j                            dd           | j        j        | j        j        |d<   | j        j        | j        j        |d<   | j        j        | j        j        |d<   | j        j        | j        j        dk    r| j        j        |d<   | j	        j
        | j	        j
        |d	<   | j        j        r| j        j        |d
<   | j        j        rd|d<   | j        j        |d<   | j                            dd          |d<   | j                            dd          |d<   |S )zget 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_kwargszUadTrainer.get_train_kwargs(   sT     1 12


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