Align make_private_with_epsilon with make_private (#509)

Summary:
Fixes #467

Pull Request resolved: #509

Reviewed By: karthikprasad

Differential Revision: D39538401

Pulled By: alexandresablayrolles

fbshipit-source-id: b7804fd8a82440937831e504e8269a2a649efbda

Author: Alex Sablayrolles

opacus/privacy_engine.py

@@ -443,11 +443,13 @@ def make_private_with_epsilon(
         target_epsilon: float,
         target_delta: float,
         epochs: int,
-        max_grad_norm: float,
+        max_grad_norm: Union[float, List[float]],
         batch_first: bool = True,
         loss_reduction: str = "mean",
+        poisson_sampling: bool = True,
+        clipping: str = "flat",
         noise_generator=None,
-        grad_sample_mode="hooks",
+        grad_sample_mode: str = "hooks",
         **kwargs,
     ):
         """
@@ -522,6 +524,8 @@ def make_private_with_epsilon(
             loss_reduction=loss_reduction,
             noise_generator=noise_generator,
             grad_sample_mode=grad_sample_mode,
+            poisson_sampling=poisson_sampling,
+            clipping=clipping,
         )
 
     def get_epsilon(self, delta):