fix differential_privacy

benmalef · Nov 22, 2024 · 16ea8db · 16ea8db
1 parent 1a713bb
commit 16ea8db
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/GANDLF/models/imagenet_unet.py b/GANDLF/models/imagenet_unet.py
@@ -253,7 +253,7 @@ def __init__(self, parameters) -> None:
         )
 
         # all BatchNorm should be replaced with InstanceNorm for DP experiments
-        if "differential_privacy" in parameters:
+        if parameters["differential_privacy"] is not None:
             self.replace_batchnorm(self.model)
 
         if self.n_dimensions == 3:

diff --git a/GANDLF/utils/pydantic_config.py b/GANDLF/utils/pydantic_config.py
@@ -74,5 +74,5 @@ class Parameters(BaseModel):
     headers: Optional[dict] = None
     output_dir: Optional[str] = ""
     problem_type: Optional[str] = None
-    differential_privacy: Optional[dict] = {}
+    differential_privacy: Optional[dict] = None
     # opt: Optional[Union[dict, str]] = {}  # TODO find a better way