mirror of
https://github.com/wassname/ray.git
synced 2026-07-18 12:40:56 +08:00
Prevent Local Worker creation from blocking remote worker creation by creating remote workers before local worker (#10245)
* create remote workers before local worker * reformatted
This commit is contained in:
@@ -64,15 +64,15 @@ class WorkerSet:
|
||||
trainer_config,
|
||||
{"tf_session_args": trainer_config["local_tf_session_args"]})
|
||||
|
||||
# Create a number of remote workers
|
||||
self._remote_workers = []
|
||||
self.add_workers(num_workers)
|
||||
|
||||
# Always create a local worker
|
||||
self._local_worker = self._make_worker(RolloutWorker, env_creator,
|
||||
self._policy_class, 0,
|
||||
self._local_config)
|
||||
|
||||
# Create a number of remote workers
|
||||
self._remote_workers = []
|
||||
self.add_workers(num_workers)
|
||||
|
||||
def local_worker(self) -> RolloutWorker:
|
||||
"""Return the local rollout worker."""
|
||||
return self._local_worker
|
||||
|
||||
Reference in New Issue
Block a user