o
    `۷i                  
   @   s   d Z ddlZddlZddlmZ dd Zdd ZedkrpddlZe Z	e	j
d	d
dd e	 \ZZejdd ejeejddejrDdnddejrLdndeddeddeddgddZe Zede j dS dS )zGThis example demonstrates basic Ray Tune random search and grid search.    N)tunec                 C   s&   t d d||  d  d |d  S )Ng?d   )timesleep)stepwidthheight r
   Z/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/tune/examples/tune_basic_example.pyevaluation_fn   s   
r   c                 C   sD   | d | d }}t | d D ]}t|||}t||d qd S )Nr   r	   steps)
iterations	mean_loss)ranger   r   report)configr   r	   r   intermediate_scorer
   r
   r   easy_objective   s
   r   __main__z--smoke-test
store_truezFinish quickly for testing)actionhelpF)configure_loggingr   min   2   )metricmodenum_samplesr      irelutanh)r   r   r	   
activation)tune_configparam_spacez!Best hyperparameters found were: )__doc__r   rayr   r   r   __name__argparseArgumentParserparseradd_argumentparse_known_argsargs_initTuner
TuneConfig
smoke_testuniformgrid_searchtunerfitresultsprintget_best_resultr   r
   r
   r
   r   <module>   s<    

