Aleksandr Borzunov 3 gadi atpakaļ
vecāks
revīzija
69d0a94e72
1 mainītis faili ar 0 papildinājumiem un 1 dzēšanām
  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