Update hivemind/optim/experimental/state_averager.py

Co-authored-by: Alexander Borzunov <[email protected]>
learning-at-home · Nov 15, 2021 · 0ee16c8 · 0ee16c8
1 parent 7074785
commit 0ee16c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hivemind/optim/experimental/state_averager.py b/hivemind/optim/experimental/state_averager.py
@@ -516,7 +516,7 @@ def load_state_from_peers(self, **kwargs):
         try:
             load_optimizer_state(self.optimizer, metadata["optimizer_metadata"], loaded_opt_tensors)
         except StopIteration:
-            logger.error("Failed to load state from peer, received inconsistent number of optimizer statistics.")
+            logger.warning("Failed to load state from peer, received inconsistent number of optimizer statistics")
             return
 
         with torch.no_grad():