Aleksandr Borzunov 3 年之前
父節點
當前提交
69d0a94e72
共有 1 個文件被更改,包括 0 次插入1 次删除
  1. 0 1
      hivemind/optim/experimental/optimizer.py

+ 0 - 1
hivemind/optim/experimental/optimizer.py

@@ -206,7 +206,6 @@ class Optimizer(torch.optim.Optimizer):
         closure: Optional[Callable[[], torch.Tensor]] = None,
         batch_size: Optional[int] = None,
         grad_scaler: Optional[GradScaler] = None,
-        **kwargs,
     ):
         """
         Report accumulating gradients w.r.t. batch_size additional samples, optionally update model parameters