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 )
ÚOVSegTrainerz*Open Vocabulary Segmentation Model Trainerc                 C   s   t dƒ‚)NzAopen vocabulary segmentation models do not support train for now.r   )ÚselfÚconfig© r   úu/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/open_vocabulary_segmentation/trainer.pyÚ__init__   s   ÿzOVSegTrainer.__init__c                 C   s   dS )zupdate training configNr   )r	   r   r   r   Úupdate_config   s    zOVSegTrainer.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OVSegTrainer.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   