
    ~jr                     n    d dl mZ d dlmZmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ  G d de          Zd	S )
   )_ModelBasedConfig)
get_config
parse_args)raise_unsupported_api_error)INFER_BENCHMARK)disable_pir_bydefault)try_except_decoratorc                   .    e Zd Zd Zd Zed             ZdS )Enginec                     t                      }t          |j        |j        d          }|j        j        | _        |j        j        | _        t          |          | _
        d S )NF)	overridesshow)r   r   configoverrideGlobalmode_modeoutput_outputr   _model)selfargsr   s      ^/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddlex/engine.py__init__zEngine.__init__   sN    ||DK4=uMMM]'
}+'//    c                    | j         dk    r| j                                        S | j         dk    r)t                       | j                                         dS | j         dk    r't                       | j                                        S | j         dk    r't                       | j                                        S | j         dk    r| j                                        S | j         dk    r\| j                                        D ]@}t          r
|
                                 | j        r|                    | j                   AdS t          | j          | j                   dS )	zthe main functioncheck_datasettrainevaluateexportpdparams2safetensorspredict)	save_pathN)r   r   r   r   r   r   r    r!   r"   r   printr   save_allr   	__class__)r   ress     r   runz
Engine.run"   s_    :((;,,...Z7""!###KZ:%%!###;'')))Z8##!###;%%'''Z111;33555Z9$${**,, 9 9" 		< 9LL4<L8889 9 (4:HHHHHr   N)__name__
__module____qualname____doc__r   r	   r(    r   r   r   r      sI        L0 0 0 I I I I Ir   r   N)modelr   utils.configr   r   utils.errorsr   utils.flagsr   utils.lazy_loaderr   utils.result_saverr	   objectr   r-   r   r   <module>r5      s     % $ $ $ $ $ 0 0 0 0 0 0 0 0 5 5 5 5 5 5 ( ( ( ( ( ( 4 4 4 4 4 4 4 4 4 4 4 4"I "I "I "I "IV "I "I "I "I "Ir   