
    |-j                     :    d dl mZ ddlmZ  G d de          ZdS )   )BaseEvaluator   )MODELSc                   (    e Zd ZdZeZd ZdefdZdS )ClsEvaluatorz$Image Classification Model Evaluatorc                    | j         j        r$| j                            | j         j                   | j        d         d         dk    r6| j                            d           | j                            d           | j                            | j        j        d           | j        	                    | j         j
                   dS )zupdate evaluation configArchnameDistillationModelF)
pretrained
ClsDatasetN)eval_configlog_interval
pdx_configupdate_log_intervalupdate_teacher_modelupdate_student_modelupdate_datasetglobal_configdataset_dirupdate_pretrained_weightsweight_pathselfs    n/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/modules/image_classification/evaluator.pyupdate_configzClsEvaluator.update_config   s    ( 	OO//0@0MNNN?6"6*.AAAO00E0BBBO00E0BBB&&t'9'E|TTT11$2B2NOOOOO    returnc                 H    | j         j        |                     d          dS )zget key-value arguments of model evaluation function

        Returns:
            dict: the arguments of evaluation function.
        r   )using_device_number)r   device)r   r   
get_devicer   s    r   get_eval_kwargszClsEvaluator.get_eval_kwargs"   s-      +7oo!o<<
 
 	
r   N)	__name__
__module____qualname____doc__r   entitiesr   dictr#    r   r   r   r      sL        ..HP P P	
 	
 	
 	
 	
 	
 	
r   r   N)baser   
model_listr   r   r*   r   r   <module>r-      s_    !                
 
 
 
 
= 
 
 
 
 
r   