fixed minor bug

sergiuabed · May 8, 2023 · 4a48e45 · 4a48e45
1 parent a4efb75
commit 4a48e45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/client/client.py b/client/client.py
@@ -82,7 +82,7 @@ def private_train(self):
 
         # Define optimizer
         optimizer = optim.SGD(  
-            self._model.parameters(), lr=LR_SGD, momentum=None, weight_decay=WEIGHT_DECAY
+            self._model.parameters(), lr=LR_SGD, weight_decay=WEIGHT_DECAY
         )
         LR = LR_SGD
         #optimizer = self.consensus_optimizer #parameters to optimize already passed during the init of the client