changed params

aidotse · Sep 29, 2023 · 10a7a2b · 10a7a2b
1 parent a2b77bc
commit 10a7a2b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/federated-learning-v2/main.py b/federated-learning-v2/main.py
@@ -52,7 +52,7 @@ def main():
     Module = LogisticRegressor 
     Optimizer = torch.optim.SGD
     Criterion = torch.nn.CrossEntropyLoss
-    lr = 0.01
+    lr = 0.02
     n_epochs = 1 
     batch_size = 64
     n_workers = 1