Aleksandr Borzunov 3 rokov pred
rodič
commit
69d0a94e72
1 zmenil súbory, kde vykonal 0 pridanie a 1 odobranie
  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