applying comments on the handler

aidotse · Jan 18, 2025 · 272d01c · 272d01c
1 parent 080506e
commit 272d01c
Show file tree

Hide file tree

Showing 2 changed files with 123 additions and 61 deletions.
diff --git a/examples/mia/celebA_HQ/celebA_HQ_handler.py b/examples/mia/celebA_HQ/celebA_HQ_handler.py
@@ -19,10 +19,7 @@ def get_criterion(self) -> torch.nn.Module:
 
     def get_optimizer(self, model: torch.nn.Module) -> optim.Optimizer:
         """Set the optimizer for the model."""
-        learning_rate =0.01  # Default learning rate
-        momentum =  0.9  # Default momentum
-        weight_decay = 0.0001  # Default weight decay
-        return optim.SGD(model.parameters(), lr=learning_rate, momentum=momentum, weight_decay=weight_decay)
+        return optim.SGD(model.parameters())
 
     def train(
         self,