o "jo @sddlZddlmZmZddlmZiddefddefdd efd d efd d efddefddefddefddefddefddefddefddefddefd d!efd"d#efd$d%efd&efd'efd(efd)efd*efd+efd,efd-Zd.d/Z d0d1Z dS)2N) REMAINDERArgumentParser) strtoboolZPOD_IPhostZ PADDLE_MASTERZmasterZPADDLE_DEVICESZdevicesZ PADDLE_NNODESZnnodesZPADDLE_RUN_MODEZrun_modeZPADDLE_LOG_LEVELZ log_levelZPADDLE_LOG_OVERWRITEZ log_overwriteZPADDLE_SORT_IPZsort_ipZPADDLE_NPROC_PER_NODEZnproc_per_nodeZ PADDLE_JOB_IDZjob_idZ PADDLE_RANKrankZPADDLE_LOG_DIRZlog_dirZPADDLE_MAX_RESTARTZ max_restartZPADDLE_ELASTIC_LEVELZ elastic_levelZPADDLE_ELASTIC_TIMEOUTZelastic_timeoutZPADDLE_SERVER_NUMZ server_numZPADDLE_TRAINER_NUMZ trainer_numZserversZtrainersZ gloo_portZ with_glooZ start_portipsZauto_parallel_config)ZPADDLE_SERVERS_ENDPOINTSZPADDLE_TRAINERS_ENDPOINTSZPADDLE_GLOO_PORTZPADDLE_WITH_GLOOZPADDLE_START_PORTZ PADDLE_IPSZPADDLE_AUTO_PARALLEL_CONFIGcCs&tjddtjddtjS)NZ http_proxyZ https_proxy)osenvironpopcopyr r l/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/distributed/launch/context/args_envs.py fetch_envs/s rc Csrt}|d}|jdtddd|jdtddd|jdtd d d|jd td d d|jdtddd|jdtddd|jdtddd|jdtddd|jdtddd|jdtddd|jdtddd|jdtd d!d|jd"d#d$d%tdd&d|jd'tdd(d|jd)tdd*d|jd+td,d-d|jd.tdd/d|jd0td1d2|jd3tdd4d|jd5td6|d7}|jd8td9d:d|jd;td9dd|jd?tdd@d|jdAtdBdCd|jdDtddEd|dF}|jdGtdHdId|jdJtd dKd|jdLtdMdNd|}tt dOd }|dPkr7t |dPdQs2J||dP_ |S)RNzBase Parametersz--masterz%the master/rendezvous server, ip:port)typedefaulthelpz--legacyFzuse legacy launchz--rankz the node rankz --log_levelINFOzlog level. Default INFOz--log_overwritez'overwrite exits logfiles. Default Falsez --sort_ipzrank node by ip. Default Falsez--enable_gpu_logTz2enable capture gpu log while running. Default Truez--nnodes1z)the number of nodes, i.e. pod/node numberz--nproc_per_nodez the number of processes in a podz --log_dirlogz.the path for each process's log. Default ./logz --run_modez+run mode of the job, collective/ps/ps-heterz--job_idrz%unique id of the job. Default defaultz --devicesz--gpusz--npusz--xpusz'accelerate devices. as --gpus,npus,xpusz--hostzhost ipz--ipsz#nodes ips, e.g. 10.10.1.1,10.10.1.2z --start_portizfix port start withz--auto_parallel_configzGauto parallel config file absolute path, the file should be json formatZtraining_scriptzHthe full path of py script,followed by arguments for the training script)rrz--auto_tuner_jsonzauto tuner json file pathZtraining_script_args)nargszParameter-Server Parametersz --serverszservers endpoints full listz --trainersztrainers endpoints full listz --trainer_numznumber of trainersz --server_numznumber of serversz --gloo_portiozgloo http portz --with_gloozuse gloo or notzElastic Parametersz --max_restartz the times can restart. Default 3z--elastic_levelzHelastic level: -1 disable, 0 failed exit, peers hold, 1 internal restartz--elastic_timeoutz1seconds to wait before elastic job begin to trainZPADDLE_TRAINER_IDrr) radd_argument_group add_argumentstrrintrparse_known_argsrgetenvhasattrr)parserZ base_groupZps_groupZ elastic_groupargsZenv_rankr r r parse_args6s$      r#) rargparserrZdistutils.utilrrrZenv_args_mappingrr#r r r r s`