浏览代码

fix kwargs

xtinkt 4 年之前
父节点
当前提交
3eed4e91c7
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      hivemind/optim/averaged.py

+ 1 - 1
hivemind/optim/averaged.py

@@ -119,7 +119,7 @@ class DecentralizedOptimizer(DecentralizedOptimizerBase):
     def _sync_if_needed(self):
         if not self.is_synchronized():
             logger.warning("Peer is out of sync.")
-            self.load_states_from_peers(**kwargs)
+            self.load_states_from_peers()
             return
 
     def zero_grad(self, *args, **kwargs):