Jelajahi Sumber

fix asserts

Artem Chumachenko 3 tahun lalu
induk
melakukan
4362b6be4f
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      hivemind/optim/power_sgd_averager.py

+ 1 - 1
hivemind/optim/power_sgd_averager.py

@@ -233,7 +233,7 @@ class PowerSGDGradientAverager(GradientAverager):
         metadata, flat_tensors = loaded_state
         logger.info("Starting loading gradient averager buffers from peers")
 
-        if num_parameters_and_extras != len(self._qs):
+        if len(flat_tensors) != len(self._qs):
             logger.error("Failed to load state from peer, received parameters, extras or metadata")
             return