Răsfoiți Sursa

is the new black

justheuristic 3 ani în urmă
părinte
comite
ac8a1a4308
1 a modificat fișierele cu 2 adăugiri și 1 ștergeri
  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")