Skip to content

Commit f31d3b1

Browse files
authored
manager_integ_tests: added LocalSGD integration test (#55)
1 parent a32f807 commit f31d3b1

File tree

2 files changed

+205
-106
lines changed

2 files changed

+205
-106
lines changed

torchft/local_sgd.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ def _average(self) -> None:
178178

179179
for p in self._model.parameters():
180180
# TODO: bucketize parameters
181-
works.append(self._manager.allreduce(p))
181+
works.append(self._manager.allreduce(p.data.detach()))
182182

183183
for work in works:
184184
work.wait()

0 commit comments

Comments
 (0)