← Về thư mục
verl/single_controller/base/worker.py
- DistRankInfo · class · L24-L27 — class DistRankInfo
- DistGlobalInfo · class · L31-L34 — class DistGlobalInfo
- WorkerHelper · class · L37-L67 — class WorkerHelper
- _get_node_ip · method · L39-L56 — def _get_node_ip(self)
- get_node_ip_by_sdk · function · L41-L48 — def get_node_ip_by_sdk()
- _get_free_port · method · L58-L61 — def _get_free_port(self)
- get_availale_master_addr_port · method · L63-L64 — def get_availale_master_addr_port(self)
- _get_pid · method · L66-L67 — def _get_pid(self)
- WorkerMeta · class · L70-L79 — class WorkerMeta
- init · method · L75-L76 — def init(self, store) -> None
- to_dict · method · L78-L79 — def to_dict(self)
- Worker · class · L83-L186 — class Worker(WorkerHelper)
- new · method · L85-L100 — def new(cls, args, *kwargs)
- _configure_before_init · method · L102-L117 — def _configure_before_init(self, register_center_name: str, rank: int)
- init · method · L119-L145 — def init(self, cuda_visible_devices=None) -> None: # construct a meta from envrionment variable. Note that the import must be inside the class because it is executed remotely
- _configure_with_meta · method · L147-L160 — def _configure_with_meta(self, meta: WorkerMeta)
- get_master_addr_port · method · L162-L163 — def get_master_addr_port(self)
- get_cuda_visible_devices · method · L165-L168 — def get_cuda_visible_devices(self)
- world_size · method · L171-L172 — def world_size(self)
- rank · method · L175-L176 — def rank(self)
- execute_with_func_generator · method · L179-L181 — def execute_with_func_generator(self, func, args, *kwargs)
- execute_func_rank_zero · method · L184-L186 — def execute_func_rank_zero(self, func, args, *kwargs)