Michael Diskin před 4 roky
rodič
revize
d6e808687a
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      hivemind/averaging/training.py

+ 1 - 1
hivemind/averaging/training.py

@@ -91,7 +91,7 @@ class TrainingAverager(DecentralizedAverager):
                     averaged_tensor[...] = local_tensor.cpu().float()
             logger.debug(f"AAAAA 5")
             self.pending_updates_done.set()
-        logger.debug(f"AAAAA 84")
+            logger.debug(f"AAAAA 84")
 
             # find a group and hopefully average tensors with peers, use batch sizes as weights
             gathered = super().step(**kwargs)