Michael Diskin 4 years ago
parent
commit
d6e808687a
1 changed files with 1 additions and 1 deletions
  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)