소스 검색

fix asserts

Artem Chumachenko 3 년 전
부모
커밋
4362b6be4f
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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