Fix a bug about a missing argument

DPBayes · Oct 4, 2024 · a1b55fe · a1b55fe
1 parent 3140541
commit a1b55fe
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/opacus/privacy_engine.py b/opacus/privacy_engine.py
@@ -139,6 +139,7 @@ def _prepare_optimizer(
             generator=generator,
             secure_mode=self.secure_mode,
             normalize_clipping=normalize_clipping,
+            optim_args=optim_args,
             **kwargs,
         )