wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-15 11:25:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Ray SGD] [Hotfix] Worker group hotfix (#11008)

Browse Source
This commit is contained in:
Amog Kamsetty
2020-09-24 12:21:30 -07:00
committed by GitHub
parent 8c241d5f1d
commit 07bdf062b9
2 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/sgd/torch/worker_group.py
+1 -1
View File
@@ -464,7 +464,7 @@ class LocalWorkerGroup(WorkerGroupInterface):
return [local_call] + ray.get(remote_calls)
def apply_all_workers(self, fn):
remote_calls = self.remote_worker_group.apply_all_workers(fn)
remote_calls = self.remote_worker_group._apply_all_workers(fn)
local_call = self.local_worker.apply(fn)
return [local_call] + ray.get(remote_calls)