o
    1Îj°  ã                   @   s8   d dl mZ ddlmZ ddlmZ G dd„ deƒZdS )	é   ©ÚUnsupportedAPIErroré   )ÚBaseTraineré   )ÚMODELSc                   @   s2   e Zd ZdZeZdd„ Zdd„ Zdefdd„Z	d	S )
ÚOVDetTrainerz'Open Vocabulary Detection Model Trainerc                 C   s   t dƒ‚)Nz>open vocabulary detection models do not support train for now.r   )ÚselfÚconfig© r   úr/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/open_vocabulary_detection/trainer.pyÚ__init__   s   ÿzOVDetTrainer.__init__c                 C   s   dS )zupdate training configNr   )r	   r   r   r   Úupdate_config   s    zOVDetTrainer.update_configÚreturnc                 C   s   |   ¡ | j dd¡dœ}|S )zƒget key-value arguments of model training function

        Returns:
            dict: the arguments of training function.
        ÚampZOFF)Zdevicer   )Z
get_deviceZtrain_configÚget)r	   Z
train_argsr   r   r   Úget_train_kwargs"   s   þzOVDetTrainer.get_train_kwargsN)
Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   Úentitiesr   r   Údictr   r   r   r   r   r      s    r   N)Zutils.errorsr   Úbaser   Z
model_listr   r   r   r   r   r   Ú<module>   s   