o
    "Õj”  ã                   @   sþ  d dl mZ d dlmZ d dlmZmZmZ d dlm	Z	m
Z
 d dlmZ d dlmZmZ d dlmZmZmZmZmZ d dlmZ d d	lmZ d d
lmZ d dlmZ d dlmZ g d¢Z eƒ Zej!Z!ej"Z"ej#Z#ej$Z$ej%Z%ej&Z&ej'Z'ej(Z(ej&Z)ej'Z*ej'Z+ej,Z,ej-Z-ej.Z.e.Z/ej0Z0ej1Z1ej2Z2ej3Z3ej4Z4ej5Z5ej6Z6ej7Z7ej8Z8ej9Z9eƒ Z:ej;Z;ej<Z<ej=Z=ej>Z>ej?Z?ej@Z@eZejAZAejBZBejCZCejDZDejEZEejFZFejGZGejHZHejIZIejJZJeZejKZKejLZLeZejMZMejNZNejOZOejPZPejQZQddl mRZS dS )é   )Úmetrics)ÚDistributedStrategy)ÚPaddleCloudRoleMakerÚRoleÚUserDefinedRoleMaker)ÚCommunicateTopologyÚHybridCommunicateGroup)ÚUtilBase)ÚMultiSlotDataGeneratorÚMultiSlotStringDataGenerator)ÚBoxPSDatasetÚDatasetBaseÚFileInstantDatasetÚInMemoryDatasetÚQueueDataset)ÚFleet)Údistributed_model)Údistributed_optimizer)Údistributed_scaler)Úlog_util)
r   r	   r   r   r   r   r   r
   r   r   é   )Úauto_parallelN)TÚ r   Zbase.distributed_strategyr   Zbase.role_makerr   r   r   Zbase.topologyr   r   Zbase.util_factoryr	   Zdata_generator.data_generatorr
   r   Zdatasetr   r   r   r   r   Zfleetr   Úmodelr   Z	optimizerr   Zscalerr   Úutilsr   Ú__all__Z_final_strategyZ_get_applied_meta_listZ_get_applied_graph_listÚinitZis_first_workerZworker_indexZ
worker_numZnode_numÚrankZnranksZ
world_sizeZlocal_device_idsZworld_device_idsZ
local_rankZrank_in_nodeZ	is_workerZis_coordinatorZinit_coordinatorZmake_fl_strategyZget_fl_clientZworker_endpointsZ
server_numZserver_indexZserver_endpointsZ	is_serverÚutilZbarrier_workerZ
all_reduceZinit_workerZinit_serverZ
run_serverZstop_workerZsave_inference_modelZsave_persistablesZsave_cache_modelZcheck_save_pre_patch_doneZsave_one_tableZsave_dense_paramsZ
load_modelZload_inference_modelZload_one_tableZminimizeÚshrinkZget_hybrid_communicate_groupZset_log_levelZget_log_level_codeZget_log_level_nameZsave_cache_tableZcollective_perfr   Úauto© r!   r!   úb/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/distributed/fleet/__init__.pyÚ<module>   s†   