o 1jl@s,ddlmZddlmZGdddeZdS)) BaseEvaluator)MODELSc@s*eZdZdZeZddZdefddZdS)VideoClsEvaluatorz$Image Classification Model EvaluatorcCs^|jjr |j|jj|j|jjd|jjdur%|jj|jjdd|j |jj dS)zupdate evaluation configZVideoClsDatasetNeval)mode) eval_configZ log_intervalZ pdx_configZupdate_log_intervalZupdate_datasetZ global_configZ dataset_dirZ batch_sizeZupdate_batch_sizeZupdate_pretrained_weights weight_pathselfr o/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/video_classification/evaluator.py update_configs zVideoClsEvaluator.update_configreturncCs|jj|jdddS)zget key-value arguments of model evaluation function Returns: dict: the arguments of evaluation function. r)Zusing_device_number)r Zdevice)rr Z get_devicer r r r get_eval_kwargs#s z!VideoClsEvaluator.get_eval_kwargsN) __name__ __module__ __qualname____doc__rentitiesrdictrr r r r rs  rN)baserZ model_listrrr r r r s