Pārlūkot izejas kodu

is the new black

justheuristic 3 gadi atpakaļ
vecāks
revīzija
ac8a1a4308
1 mainītis faili ar 2 papildinājumiem un 1 dzēšanām
  1. 2 1
      hivemind/optim/collaborative.py

+ 2 - 1
hivemind/optim/collaborative.py

@@ -255,7 +255,8 @@ class CollaborativeOptimizer(DecentralizedOptimizerBase):
                 weight = self.local_samples_accumulated / mean_samples_per_worker
                 try:
                     group_info = self.averager.step(
-                        weight=weight, gather=current_step, timeout=self.averaging_timeout, **kwargs)
+                        weight=weight, gather=current_step, timeout=self.averaging_timeout, **kwargs
+                    )
                     if group_info:
                         logger.log(self.status_loglevel, f"Averaged tensors successfully with {len(group_info)} peers")