o
    1 i                  	   @   s  d dl Z d dlZd dlmZ d dlmZ d dlmZ edkre  Z	e	j
dddd	 e	 \ZZejejr7d
ndd edddZejeejddejrMdndidejdddejejr^dndddededded dddZe Zede j dS dS )    N)tune)HyperBandScheduler)MyTrainableClass__main__z--smoke-test
store_truezFinish quickly for testing)actionhelp   )Znum_cpusZtraining_iteration   )Z	time_attrZmax_tZhyperband_test   T)Z	fail_fast)namestopverboseZfailure_config   Zepisode_reward_meanmax)Znum_samplesZmetricmodeZ	scheduler
   Z   d   )widthheight)Z
run_configZtune_configZparam_spacez!Best hyperparameters found were: )argparseZrayr   Zray.tune.schedulersr   Zray.tune.utils.mock_trainabler   __name__ArgumentParserparseradd_argumentparse_known_argsargs_initZ
smoke_testZ	hyperbandZTunerZ	RunConfigZFailureConfigZ
TuneConfigrandintZtunerfitresultsprintZget_best_resultconfig r%   r%   o/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/tune/examples/hyperband_example.py<module>   sD   