o 1j@s,ddlmZddlmZGdddeZdS)) BaseEvaluator)MODELSc@s*eZdZdZeZddZdefddZdS)MLClsEvaluatorz$Image Classification Model EvaluatorcCsl|jjr |j|jj|jdddkr#|jjdd|jjdd|j|jjd|j |jj dS)zupdate evaluation configZArchnameZDistillationModelF)Z pretrainedZ MLClsDatasetN) eval_configZ log_intervalZ pdx_configZupdate_log_intervalZupdate_teacher_modelZupdate_student_modelZupdate_datasetZ global_configZ dataset_dirZupdate_pretrained_weights weight_pathselfr t/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/multilabel_classification/evaluator.py update_configszMLClsEvaluator.update_configreturncCs|jj|jdddS)zget key-value arguments of model evaluation function Returns: dict: the arguments of evaluation function. r)Zusing_device_number)rZdevice)rrZ get_devicer r r r get_eval_kwargs"s zMLClsEvaluator.get_eval_kwargsN) __name__ __module__ __qualname____doc__rentitiesr dictrr r r r rs  rN)baserZ model_listrrr r r r s