o
    $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rve  Z	e	j
dddd	 e	 \ZZejr9ejd
d dZededddgidZejeejddddiejdddejedddddddeddZe Zede j dS dS )     N)tune)pbt_function)PB2__main__z--smoke-test
store_truezFinish quickly for testing)actionhelp   )num_cpus   training_iterationlrg-C6?g{Gz?)	time_attrperturbation_intervalhyperparam_boundspbt_testF   T)	fail_fast)nameverbosestopfailure_configmean_accuracymax   )	schedulermetricmodenum_samplesreuse_actors   )r   some_other_factorcheckpoint_interval)
run_configtune_configparam_spacez!Best hyperparameters found were: )argparserayr   ray.tune.examples.pbt_functionr   ray.tune.schedulers.pb2r   __name__ArgumentParserparseradd_argumentparse_known_argsargs_
smoke_testinitr   pbtTuner	RunConfigFailureConfig
TuneConfigtunerfitresultsprintget_best_resultconfig r>   r>   Z/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/tune/examples/pb2_example.py<module>   s\   	
