o
    1 iU                  
   @   s   d dl Z d dlZd dlmZ d dlmZ G dd deZdd Zdd	 Zd
d Z	e
dkrxe  Zejdddd e \ZZeje	ejde gdejrMdndidejdddeeddeddeddddZe Zede j dS dS )    N)tune)LoggerCallbackc                   @   s   e Zd Zdd ZdS )TestLoggerCallbackc                 K   s   t d| d|  d S )NzTestLogger for trial z: )print)self	iterationZtrialstrialresultinfo r   m/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/tune/examples/logging_example.pyon_trial_result   s   z"TestLoggerCallback.on_trial_resultN)__name__
__module____qualname__r   r   r   r   r   r   
   s    r   c                 C   s   d | j| jS )Nz	{}_{}_123)formatZtrainable_nameZtrial_id)r   r   r   r   trial_str_creator   s   r   c                 C   s&   t d d||  d  d |d  S )Ng?d   )timesleep)stepwidthheightr   r   r   evaluation_fn   s   
r   c                 C   sD   | d | d }}t | d D ]}t|||}t||d qd S )Nr   r   steps)Z
iterations	mean_loss)ranger   r   report)configr   r   r   Zintermediate_scorer   r   r   easy_objective   s
   r    __main__z--smoke-test
store_truezFinish quickly for testing)actionhelpZhyperband_testZtraining_iteration   r   )name	callbacksstopr   min   )ZmetricmodeZnum_samplesZtrial_name_creatorZtrial_dirname_creator
   )r   r   r   )Z
run_configZtune_configZparam_spacezBest hyperparameters: )argparser   Zrayr   Zray.tune.loggerr   r   r   r   r    r   ArgumentParserparseradd_argumentparse_known_argsargs_ZTunerZ	RunConfigZ
smoke_testZ
TuneConfigrandintZ
loguniformZtunerfitresultsr   Zget_best_resultr   r   r   r   r   <module>   sH   

