diff --git a/opacus/privacy_engine.py b/opacus/privacy_engine.py index 558c8f8e..bdddafe4 100644 --- a/opacus/privacy_engine.py +++ b/opacus/privacy_engine.py @@ -136,7 +136,6 @@ def _prepare_optimizer( loss_reduction=loss_reduction, generator=generator, secure_mode=self.secure_mode, - **kwargs, ) def _prepare_data_loader(